remove superfluous code

blaine-rister · blaine-rister · commit 81b4fca95447 · 2025-03-17T23:13:02.000-07:00
diff --git a/torch/_inductor/codegen/simd.py b/torch/_inductor/codegen/simd.py
@@ -815,34 +815,11 @@ def prepare_indexing(
 
         return self.codegen_indexing(simp_index)
 
-    def active_range_trees(self, reorder: bool = False) -> list[IterationRangesRoot]:
-        trees = [
+    def active_range_trees(self) -> list[IterationRangesRoot]:
+        return [
             t for t in self.range_trees if not t.is_reduction or self.inside_reduction
         ]
 
-        if not reorder:
-            return trees
-
-        def tree_key(tree: IterationRangesRoot) -> int:
-            assert tree.tensor_dim is not None, f"Missing tensor_dim for tree {tree}"
-            return tree.tensor_dim
-
-        # Keep trees with tensor_dim=None in their current positions.
-        sorted_trees = sorted(
-            (tree for tree in trees if tree.tensor_dim is not None),
-            key=tree_key,
-        )
-        sorted_idx = 0
-        final_trees = []
-        for tree in trees:
-            if tree.tensor_dim is None:
-                final_trees.append(tree)
-            else:
-                final_trees.append(sorted_trees[sorted_idx])
-                sorted_idx += 1
-
-        return final_trees
-
     def codegen_indexing(self, expr: sympy.Expr) -> sympy.Expr:
         expr = V.graph.sizevars.simplify_with_ranges(expr, self.var_ranges())
         for sym in sorted(expr.free_symbols, key=str):
diff --git a/torch/_inductor/codegen/triton.py b/torch/_inductor/codegen/triton.py
@@ -1966,7 +1966,7 @@ def match_block_pointer() -> Optional[BlockPtrOptions]:
                 index_relative_to_xyr_index = sympy_subs(
                     index, {v: t.expr for v, t in self.range_tree_nodes.items()}
                 )
-                range_trees = self.active_range_trees(reorder=True)
+                range_trees = self.active_range_trees()
 
                 # Partition the index into subexpressions pertaining to each range tree.
                 # For example xindex * 5 + r0_index * 3 is partitioned to

Original file line number	Diff line number	Diff line change
`@@ -1966,7 +1966,7 @@ def match_block_pointer() -> Optional[BlockPtrOptions]:`
`1966`	`1966`	`index_relative_to_xyr_index = sympy_subs(`
`1967`	`1967`	`index, {v: t.expr for v, t in self.range_tree_nodes.items()}`
`1968`	`1968`	`)`
`1969`		`- range_trees = self.active_range_trees(reorder=True)`
	`1969`	`+ range_trees = self.active_range_trees()`
`1970`	`1970`
`1971`	`1971`	`# Partition the index into subexpressions pertaining to each range tree.`
`1972`	`1972`	`# For example xindex * 5 + r0_index * 3 is partitioned to`