Update triton_heuristics.py

jataylo · jataylo · commit 809739fe9999 · 2025-09-19T04:32:56.000-05:00
(cherry picked from commit 4377d1f)
diff --git a/torch/_inductor/runtime/triton_heuristics.py b/torch/_inductor/runtime/triton_heuristics.py
@@ -2741,7 +2741,7 @@ def _persistent_reduction_configs(
             triton_config_reduction(size_hints, xblock, rnumel, register_intensive=True)
             for xblock in (1, 8, 32, 128)
             if xblock == 1
-            or (xblock <= xnumel and (max_autotune_enabled or rnumel * xblock <= 4096))
+            or (xblock <= xnumel and rnumel * xblock <= 4096)
         ]
     else:
         configs = []

Original file line number	Diff line number	Diff line change
`@@ -2741,7 +2741,7 @@ def _persistent_reduction_configs(`
`2741`	`2741`	`triton_config_reduction(size_hints, xblock, rnumel, register_intensive=True)`
`2742`	`2742`	`for xblock in (1, 8, 32, 128)`
`2743`	`2743`	`if xblock == 1`
`2744`		`- or (xblock <= xnumel and (max_autotune_enabled or rnumel * xblock <= 4096))`
	`2744`	`+ or (xblock <= xnumel and rnumel * xblock <= 4096)`
`2745`	`2745`	`]`
`2746`	`2746`	`else:`
`2747`	`2747`	`configs = []`