Skip to content

Commit 484b9c6

Browse files
authored
[TUTORIALS] Bypass tma_ws in the tutorial when the GPU doesn't support TMA (#6284)
1 parent c4452b2 commit 484b9c6

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

python/tutorials/09-persistent-matmul.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -725,9 +725,9 @@ def bench(K, dtype, reps=1000, warmup_reps=10000):
725725
if dtype == torch.float16:
726726
bench_fn(reps, warmup_reps, torch_matmul, a, b)
727727
bench_fn(reps, warmup_reps, matmul, a, b.T)
728-
bench_fn(reps, warmup_reps, matmul_tma_ws, a, b)
729728
bench_fn(reps, warmup_reps, matmul_persistent, a, b.T)
730729
if supports_tma():
730+
bench_fn(reps, warmup_reps, matmul_tma_ws, a, b)
731731
bench_fn(reps, warmup_reps, matmul_tma_persistent, a, b)
732732
bench_fn(reps, warmup_reps, matmul_descriptor_persistent, a, b)
733733

0 commit comments

Comments
 (0)