From 41b846152ff22dfc2f1800ec82327dec4561f085 Mon Sep 17 00:00:00 2001 From: "Su, Tong" Date: Thu, 7 Nov 2024 15:04:16 +0800 Subject: [PATCH] Revert "change target[1] to sycl_device to get arch properties from Triton side (#590)" --- 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 26dd4aaf5..383fd96f6 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"],