diff --git a/.github/scripts/apply_torch_pr.py b/.github/scripts/apply_torch_pr.py index b4b441263..d48121ff6 100644 --- a/.github/scripts/apply_torch_pr.py +++ b/.github/scripts/apply_torch_pr.py @@ -12,9 +12,8 @@ # Fallback to CPU for XPU FP64 "https://github.com/pytorch/pytorch/pull/126516", # Modify the tolerance level in TIMM benchmark - "https://github.com/pytorch/pytorch/pull/129735", - # [XPU] Update XPU C Shim Header - "https://github.com/pytorch/pytorch/pull/141086", + # "https://github.com/pytorch/pytorch/pull/129735", + "https://github.com/mengfei25/pytorch/pull/21", ] ) parser.add_argument('--extra-pr-list', '-e', nargs='+',default=[])