Skip to content

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

torch/_inductor/codegen/wrapper.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2928,6 +2928,10 @@ def codegen_subgraph_call(self, subgraph, outer_inputs, outer_buffer_name):
29282928

29292929
self.writeline(f"{subgraph.graph.name}_args = [{outer_input_names}]")
29302930

2931+
# Since the buffers are already put into the args list, we can free the
2932+
# buffers here.
2933+
V.graph.scheduler.free_buffers()
2934+
29312935
# Call the subgraph launcher function
29322936
self.writeline(
29332937
f"{outer_buffer_name} = {subgraph.graph.name}({subgraph.graph.name}_args)"

0 commit comments

Comments
 (0)