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 si…"

This reverts commit a64b397.
  • Loading branch information
quintinwang5 authored Apr 17, 2024
1 parent fcd22de commit c55a218
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 c55a218

Please sign in to comment.