better convert + write-back

2023-01-04 17:08:08 -08:00
parent 29a1e20b58
commit 268d2cd18d
2 changed files with 5 additions and 5 deletions
--- a/python/tutorials/06-fused-attention.py
+++ b/python/tutorials/06-fused-attention.py
@@ -331,7 +331,7 @@ BATCH, N_HEADS, N_CTX, D_HEAD = 4, 48, 4096, 64
 # vary seq length for fixed head and batch=4
 configs = [triton.testing.Benchmark(
    x_names=['N_CTX'],
-    x_vals=[2**i for i in range(10, 15)],
+    x_vals=[2**i for i in range(10, 13)],
    line_arg='provider',
    line_vals=['triton'],
    line_names=['Triton'],