Skip to content

Commit 8bd33f9

Browse files
naromero77amdjataylo
authored andcommitted
pointwise config with MAX_BLOCK.
(cherry picked from commit 16e8266)
1 parent af5f678 commit 8bd33f9

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

torch/_inductor/runtime/triton_heuristics.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2486,6 +2486,9 @@ def pointwise(
24862486
triton_config_with_settings(
24872487
size_hints, bs // 2, num_elements_per_warp=64
24882488
),
2489+
triton_config_with_settings(
2490+
size_hints, TRITON_MAX_BLOCK["X"], waves_per_eu=2
2491+
),
24892492
*hinted_configs,
24902493
]
24912494
# Additional reduction configs appended for ROCm builds

0 commit comments

Comments
 (0)