gistfile0.txt

diff --git a/torch/_inductor/ir.py b/torch/_inductor/ir.py
index 95898fe289d..da70d0dde08 100644
--- a/torch/_inductor/ir.py
+++ b/torch/_inductor/ir.py
@@ -1506,6 +1506,9 @@ class Reduction(Loops):
             reduction_numel,
             input_node,
         )
+        print(f"split={split}")
+        if split > 1:
+            split = max(split, 256)
         # intermediate reduction in split can contain complex indexing,
         # and num_splits will fail to correctly set the hint
         # reuse the passed hint if available