Skip to content

Commit

Permalink
change target[1] to sycl_device to get arch properties from Triton side
Browse files Browse the repository at this point in the history
  • Loading branch information
quintinwang5 committed Apr 12, 2024
1 parent fa34731 commit 47015a8
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"], 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"],
Expand Down

0 comments on commit 47015a8

Please sign in to comment.