From faec6938a35ad9b8c8a5448499eda5002f025ecf Mon Sep 17 00:00:00 2001 From: Yutao Xu Date: Mon, 23 Dec 2024 13:52:32 +0800 Subject: [PATCH 1/2] Update apply_torch_pr.py --- .github/scripts/apply_torch_pr.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/.github/scripts/apply_torch_pr.py b/.github/scripts/apply_torch_pr.py index 89fa32fdf..bbe89ed7d 100644 --- a/.github/scripts/apply_torch_pr.py +++ b/.github/scripts/apply_torch_pr.py @@ -12,8 +12,7 @@ # 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", - "https://github.com/mengfei25/pytorch/pull/21", + "https://github.com/pytorch/pytorch/pull/143739", ] ) parser.add_argument('--extra-pr-list', '-e', nargs='+',default=[]) From 1e8cd7671173586963750f44cc002fd98dae5b66 Mon Sep 17 00:00:00 2001 From: Yutao Xu Date: Mon, 23 Dec 2024 14:24:33 +0800 Subject: [PATCH 2/2] Update SDPUtils.cpp --- src/ATen/native/transformers/SDPUtils.cpp | 2 ++ 1 file changed, 2 insertions(+) diff --git a/src/ATen/native/transformers/SDPUtils.cpp b/src/ATen/native/transformers/SDPUtils.cpp index db4409493..eca5f9829 100644 --- a/src/ATen/native/transformers/SDPUtils.cpp +++ b/src/ATen/native/transformers/SDPUtils.cpp @@ -4,6 +4,8 @@ namespace sdp { +using c10::array_of; + bool check_all_tensors_on_device(sdp_params const& params, bool debug) { // Check that all tensors are on the GPU device // This should be handled by the stub dispatch, but whe call