Skip to content

Commit 873d8e2

Browse files
AmdSampsajataylo
authored andcommitted
a few more configs for 2d grids (#2649)
Added two nice grid configs for the 2d pointwise kernel cases for WRT5 workload. Confirmed that they were picked up when using max autotune. (cherry picked from commit f1eac49)
1 parent f5c4bbd commit 873d8e2

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

torch/_inductor/runtime/triton_heuristics.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2440,6 +2440,8 @@ def pointwise(
24402440
triton_config_with_settings(size_hints, 64, 64), # ~8% better for fp16
24412441
triton_config_with_settings(size_hints, 256, 16),
24422442
triton_config_with_settings(size_hints, 16, 256),
2443+
triton_config_with_settings(size_hints, 128, 16), # wrt: +10% for some kernels
2444+
triton_config_with_settings(size_hints, 32, 512), # wrt: +30% for some kernels
24432445
triton_config_with_settings(size_hints, bs, 1),
24442446
triton_config_with_settings(size_hints, 1, bs),
24452447
*hinted_configs,

0 commit comments

Comments
 (0)