change target[1] to sycl_device to get arch properties from Triton si…

…de (#590)
intel · Nov 7, 2024 · 44ec1ce · 44ec1ce
1 parent 869e0b5
commit 44ec1ce
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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"], cc)
+            target = (compile_meta["device_type"], torch.xpu.device(torch.xpu.current_device()).sycl_device)
             options = {
                 "num_warps": compile_meta["num_warps"],
                 "num_stages": compile_meta["num_stages"],