Update

[ghstack-poisoned]
pytorch · eellison · May 16, 2025 · May 16, 2025 · May 16, 2025 · May 20, 2025
commit 948321e67d32739657b1613bcc8ade4a826c02e1
diff --git a/test/inductor/test_loop_ordering.py b/test/inductor/test_loop_ordering.py
@@ -885,6 +885,8 @@ def forward(permute):
             permute = torch.ops.aten.permute.default(arg0_1, [1, 0])
 
             out, code = run_and_get_code(torch.compile(forward), (permute))
+
+            self.assertEqual(out, forward(permute))
             FileCheck().check("YBLOCK").check("XBLOCK").run(code[0])
 
 

diff --git a/torch/_inductor/codegen/simd.py b/torch/_inductor/codegen/simd.py
@@ -711,7 +711,7 @@ def prepare_split_iteration_lengths(
         groups: Iterable[sympy.Expr],
         lengths: Sequence[Sequence[sympy.Expr]],
         reduction_numel: sympy.Expr = sympy.S.One,
-    ) -> Sequence[Sequence[sympy.Expr]]::
+    ) -> Sequence[Sequence[sympy.Expr]]:
         "Fill in the reduction numel of lengths if missing"
         sizevars = V.graph.sizevars
         if len(lengths[1]) == 0 and (
@@ -2130,6 +2130,7 @@ def process_node_vars(
             )
 
         # TODO, add tests, reduction splits if config.triton.tile_reductions
+        # TODO: we should ignore tiny increases in score for extra splits
         overlapping_iter_vars = (
             all_iter_vars & coalesce_analysis.coalesced_by_var.keys()
         )