fix permute_memory_format_pass (#684)

cccclai · facebook-github-bot · commit ddd1cd75e910 · 2023-10-07T11:58:51.000-07:00
Summary: Pull Request resolved: #684 Need it by tosa as they requires channel last for conv op Reviewed By: tarun292 Differential Revision: D50052821 fbshipit-source-id: bbfda46f01b226f906d29e18df726526092845b9
diff --git a/backends/example/example_backend_delegate_passes/permute_memory_formats_pass.py b/backends/example/example_backend_delegate_passes/permute_memory_formats_pass.py
@@ -69,7 +69,7 @@ def call(  # noqa: suprress function is too complex (13)
                                     },
                                 )
                                 # Insert to_dim op and it'll be the return op
-                                _ = graph_module.graph.output(exit_to_dim_op)
+                                _ = graph_module.graph.output((exit_to_dim_op,))
                                 # Remove the old return op.
                                 graph_module.graph.erase_node(exit_node)
                             # Handle the case when the pattern output is intermediate output,

Original file line number	Diff line number	Diff line change
`@@ -69,7 +69,7 @@ def call( # noqa: suprress function is too complex (13)`
`69`	`69`	`},`
`70`	`70`	`)`
`71`	`71`	`# Insert to_dim op and it'll be the return op`
`72`		`- _ = graph_module.graph.output(exit_to_dim_op)`
	`72`	`+ _ = graph_module.graph.output((exit_to_dim_op,))`
`73`	`73`	`# Remove the old return op.`
`74`	`74`	`graph_module.graph.erase_node(exit_node)`
`75`	`75`	`# Handle the case when the pattern output is intermediate output,`