Skip to content

Commit 3d716eb

Browse files
committed
pointwise config with MAX_BLOCK.
(cherry picked from commit 16e8266) (cherry picked from commit 8bd33f9)
1 parent a13015c commit 3d716eb

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
@@ -1897,6 +1897,9 @@ def pointwise(
18971897
triton_config_with_settings(
18981898
size_hints, bs // 2, num_elements_per_warp=64
18991899
),
1900+
triton_config_with_settings(
1901+
size_hints, TRITON_MAX_BLOCK["X"], waves_per_eu=2
1902+
),
19001903
*hinted_configs,
19011904
]
19021905
# Additional reduction configs appended for ROCm builds

0 commit comments

Comments
 (0)