Skip to content

Commit 44e07d7

Browse files
committed
Added a place to put more reduction configs
1 parent 47078b5 commit 44e07d7

File tree

1 file changed

+11
-5
lines changed

1 file changed

+11
-5
lines changed

torch/_inductor/runtime/triton_heuristics.py

Lines changed: 11 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -2537,14 +2537,14 @@ def make_config(x, r, num_warps=None, num_stages=1, register_intensive=False):
25372537
):
25382538
pass # skip all these cases
25392539
elif reduction_hint == ReductionHint.INNER:
2540-
return [contiguous_config]
2540+
result_configs = [contiguous_config]
25412541
elif reduction_hint == ReductionHint.OUTER:
2542-
return [outer_config]
2542+
result_configs = [outer_config]
25432543
elif reduction_hint == ReductionHint.OUTER_TINY:
2544-
return [tiny_config]
2544+
result_configs = [tiny_config]
25452545
if disable_pointwise_autotuning(inductor_meta):
2546-
return [make_config(32, 128)]
2547-
return [
2546+
result_configs = [make_config(32, 128)]
2547+
result_configs = [
25482548
contiguous_config,
25492549
outer_config,
25502550
tiny_config,
@@ -2556,6 +2556,12 @@ def make_config(x, r, num_warps=None, num_stages=1, register_intensive=False):
25562556
make_config(64, 4, num_warps=8),
25572557
]
25582558

2559+
# Additional reduction configs appended for ROCm builds
2560+
if torch.version.hip:
2561+
pass
2562+
2563+
return result_configs
2564+
25592565

25602566
def match_target_block_product(
25612567
size_hints, tiling_scores, target_block_product, min_block_size=1

0 commit comments

Comments
 (0)