Skip to content

Commit 821a208

Browse files
committed
Fix bug to renable reduction tuning
(cherry picked from commit 0e2ed1d)
1 parent 13d0455 commit 821a208

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
@@ -2474,7 +2474,10 @@ def _reduction_configs(
24742474
rnumel = get_total_reduction_numel(size_hints)
24752475

24762476
# Is max autotune enabled
2477-
max_autotune = inductor_meta.get("max_autotune") or inductor_meta.get("max_autotune_pointwise")
2477+
max_autotune_enabled = not disable_pointwise_autotuning(inductor_meta) or (
2478+
inductor_meta.get("max_autotune")
2479+
or inductor_meta.get("max_autotune_pointwise")
2480+
)
24782481

24792482
register_intensive = False
24802483
MAX_R0_BLOCK = 2048

0 commit comments

Comments
 (0)