Skip to content

Commit 4a229dd

Browse files
authored
[Bugfix] Fallback torch.accelerator.synchronize() to torch.cuda.synchronize() (#987)
* [Refactor]:Add support for torch version lower than 2.6.0 * update
1 parent fc41463 commit 4a229dd

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tilelang/profiler/bench.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -97,7 +97,7 @@ def do_bench(
9797

9898
# Initial function call and synchronization
9999
fn()
100-
torch.accelerator.synchronize()
100+
torch.cuda.synchronize()
101101

102102
# Create L2 cache flush buffer (256 MB)
103103
# Fast flush uses int32 (4 bytes), regular uses int8 (1 byte)

0 commit comments

Comments
 (0)