From 21b64e4bd2249e437c9cd915f0376c8af57b9770 Mon Sep 17 00:00:00 2001 From: Nathanael See Date: Thu, 20 Jun 2024 21:44:17 -0700 Subject: [PATCH] fix skip dim order use in test_vulkan_delegate (#4028) Summary: Pull Request resolved: https://github.com/pytorch/executorch/pull/4028 broke in D58593014 Differential Revision: D58855354 --- backends/vulkan/test/test_vulkan_delegate.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/backends/vulkan/test/test_vulkan_delegate.py b/backends/vulkan/test/test_vulkan_delegate.py index a59d72a9ca0..9e4f9803685 100644 --- 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))