Skip to content

Commit c8f6b02

Browse files
authored
Duplicate config block
1 parent 8c58805 commit c8f6b02

File tree

1 file changed

+1
-7
lines changed

1 file changed

+1
-7
lines changed

torch/_inductor/runtime/triton_heuristics.py

Lines changed: 1 addition & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -2740,18 +2740,12 @@ def _persistent_reduction_configs(
27402740
or inductor_meta.get("max_autotune_pointwise")
27412741
)
27422742

2743-
configs = [
2744-
triton_config_reduction(size_hints, xblock, rnumel, register_intensive=True)
2745-
for xblock in (1, 8, 32, 128)
2746-
if xblock == 1 or (xblock <= xnumel and (max_autotune_enabled or rnumel * xblock <= 4096))
2747-
]
2748-
27492743
if "y" not in size_hints:
27502744
configs = [
27512745
triton_config_reduction(size_hints, xblock, rnumel, register_intensive=True)
27522746
for xblock in (1, 8, 32, 128)
27532747
if xblock == 1
2754-
or (rnumel * xblock <= 4096 and xblock <= xnumel)
2748+
or (xblock <= xnumel and (max_autotune_enabled or rnumel * xblock <= 4096))
27552749
]
27562750
else:
27572751
configs = []

0 commit comments

Comments
 (0)