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"],