Skip to content

Commit 765ede7

Browse files
authored
Fix bug to renable reduction tuning
1 parent d09906a commit 765ede7

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

torch/_inductor/runtime/triton_heuristics.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2564,7 +2564,10 @@ def _reduction_configs(
25642564
rnumel = get_total_reduction_numel(size_hints)
25652565

25662566
# Is max autotune enabled
2567-
max_autotune = inductor_meta.get("max_autotune") or inductor_meta.get("max_autotune_pointwise")
2567+
max_autotune_enabled = not disable_pointwise_autotuning(inductor_meta) or (
2568+
inductor_meta.get("max_autotune")
2569+
or inductor_meta.get("max_autotune_pointwise")
2570+
)
25682571

25692572
register_intensive = False
25702573
MAX_R0_BLOCK = 2048

0 commit comments

Comments
 (0)