File tree Expand file tree Collapse file tree 1 file changed +14
-14
lines changed Expand file tree Collapse file tree 1 file changed +14
-14
lines changed Original file line number Diff line number Diff line change @@ -2595,20 +2595,20 @@ def _persistent_reduction_configs(
25952595 elif reduction_hint == ReductionHint .OUTER :
25962596 configs = configs [- 1 :]
25972597
2598- if reduction_hint == ReductionHint . OUTER_TINY :
2599- tiny_configs = [
2600- triton_config_reduction (
2601- size_hints ,
2602- 2 * ( 256 // rnumel ) if rnumel <= 256 else 1 ,
2603- rnumel ,
2604- )
2605- ]
2606- if max_autotune_enabled :
2607- for tconfig in tiny_configs :
2608- if tconfig not in configs :
2609- configs .append (tconfig )
2610- else :
2611- configs = tiny_configs
2598+ tiny_configs = [
2599+ triton_config_reduction (
2600+ size_hints ,
2601+ 2 * ( 256 // rnumel ) if rnumel <= 256 else 1 ,
2602+ rnumel ,
2603+ )
2604+ ]
2605+
2606+ if max_autotune_enabled :
2607+ for conf in tiny_configs :
2608+ if conf not in configs :
2609+ configs .append (conf )
2610+ elif reduction_hint == ReductionHint . OUTER_TINY :
2611+ configs = tiny_configs
26122612
26132613 for c in configs :
26142614 # we don't need Rn_BLOCK for persistent reduction
You can’t perform that action at this time.
0 commit comments