From c55a218c2c3910ac43883c5a0f00fd5159fcc5c0 Mon Sep 17 00:00:00 2001 From: "Wang, Quintin" Date: Wed, 17 Apr 2024 16:49:21 +0800 Subject: [PATCH] =?UTF-8?q?Revert=20"change=20target[1]=20to=20sycl=5Fdevi?= =?UTF-8?q?ce=20to=20get=20arch=20properties=20from=20Triton=20si=E2=80=A6?= =?UTF-8?q?"?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit This reverts commit a64b397a79177ba80a11df1791d98e6b57b77c61. --- intel_extension_for_pytorch/_inductor/xpu/triton_heuristics.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/intel_extension_for_pytorch/_inductor/xpu/triton_heuristics.py b/intel_extension_for_pytorch/_inductor/xpu/triton_heuristics.py index d604f4e3b..8c7a7ee13 100644 --- a/intel_extension_for_pytorch/_inductor/xpu/triton_heuristics.py +++ b/intel_extension_for_pytorch/_inductor/xpu/triton_heuristics.py @@ -128,7 +128,7 @@ def _precompile_config(self, cfg: Config, warm_cache_only_with_cc: Dict): ), ) - target = (compile_meta["device_type"], torch.xpu.device(torch.xpu.current_device()).sycl_device) + target = (compile_meta["device_type"], cc) options = { "num_warps": compile_meta["num_warps"], "num_stages": compile_meta["num_stages"],