enable pipeline

xiaohuguo2023 · naromero77amd · commit 4b623331719c · 2025-08-23T22:54:25.000Z
diff --git a/torch/_inductor/codegen/triton.py b/torch/_inductor/codegen/triton.py
@@ -3599,7 +3599,7 @@ def codegen_body(self):
                         "rsplit_end" if self.cooperative_reduction else f"{prefix}numel"
                     )
                     self.body.writeline(
-                        f"for {prefix}offset in range({loop_start}, {loop_end}, {prefix.upper()}BLOCK):"
+                        f"for {prefix}offset in tl.range({loop_start}, {loop_end}, {prefix.upper()}BLOCK, num_stages = 2):"
                     )
                 with self.body.indent(offset=level + 1):
                     self.iteration_ranges_codegen_header(tree, self.body)

Original file line number	Diff line number	Diff line change
`@@ -3599,7 +3599,7 @@ def codegen_body(self):`
`3599`	`3599`	`"rsplit_end" if self.cooperative_reduction else f"{prefix}numel"`
`3600`	`3600`	`)`
`3601`	`3601`	`self.body.writeline(`
`3602`		`- f"for {prefix}offset in range({loop_start}, {loop_end}, {prefix.upper()}BLOCK):"`
	`3602`	`+ f"for {prefix}offset in tl.range({loop_start}, {loop_end}, {prefix.upper()}BLOCK, num_stages = 2):"`
`3603`	`3603`	`)`
`3604`	`3604`	`with self.body.indent(offset=level + 1):`
`3605`	`3605`	`self.iteration_ranges_codegen_header(tree, self.body)`