Skip to content

Commit

Permalink
Revert "change target[1] to sycl_device to get arch properties from T…
Browse files Browse the repository at this point in the history
…riton side (#590)"
  • Loading branch information
Stonepia committed Nov 7, 2024
1 parent b3db0ed commit 41b8461
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -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"],
Expand Down

0 comments on commit 41b8461

Please sign in to comment.