fix skip dim order use in test_vulkan_delegate (#4028)

nathanaelsee · facebook-github-bot · commit 524dd490a698 · 2024-06-20T23:19:46.000-07:00
Summary: Pull Request resolved: #4028 broke in D58593014 Reviewed By: copyrightly Differential Revision: D58855354 fbshipit-source-id: c2aab78ac35e84a5c50de86d00e939f01fd6e621
diff --git a/backends/vulkan/test/test_vulkan_delegate.py b/backends/vulkan/test/test_vulkan_delegate.py
@@ -123,7 +123,9 @@ def run_test(memory_layout):
                 program, compile_config=self._edge_compile_config
             )
 
-            edge_program = edge_program.transform([I64toI32(), MeanToSumDiv()])
+            edge_program = edge_program.transform(
+                [I64toI32(self._edge_compile_config._skip_dim_order), MeanToSumDiv()]
+            )
 
             edge_program = edge_program.to_backend(VulkanPartitioner(compile_options))
 

Original file line number	Diff line number	Diff line change
`@@ -123,7 +123,9 @@ def run_test(memory_layout):`
`123`	`123`	`program, compile_config=self._edge_compile_config`
`124`	`124`	`)`
`125`	`125`
`126`		`- edge_program = edge_program.transform([I64toI32(), MeanToSumDiv()])`
	`126`	`+ edge_program = edge_program.transform(`
	`127`	`+ [I64toI32(self._edge_compile_config._skip_dim_order), MeanToSumDiv()]`
	`128`	`+ )`
`127`	`129`
`128`	`130`	`edge_program = edge_program.to_backend(VulkanPartitioner(compile_options))`
`129`	`131`