From 6a4f60d60f1055452c2ca4545f817978031d8264 Mon Sep 17 00:00:00 2001 From: Gasoonjia Date: Wed, 11 Dec 2024 00:42:15 -0800 Subject: [PATCH] [dim order] reenbale dim order as default Differential Revision: D66970153 Pull Request resolved: https://github.com/pytorch/executorch/pull/7252 (cherry picked from commit f2b299ca0a1999b458461aeb7a31543330f26f36) --- backends/vulkan/test/test_vulkan_delegate.py | 1 + exir/capture/_config.py | 3 +-- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/backends/vulkan/test/test_vulkan_delegate.py b/backends/vulkan/test/test_vulkan_delegate.py index 2816dd1020..ca54918970 100644 --- a/backends/vulkan/test/test_vulkan_delegate.py +++ b/backends/vulkan/test/test_vulkan_delegate.py @@ -117,6 +117,7 @@ def run_test(): edge_program = to_edge_transform_and_lower( program, + compile_config=self._edge_compile_config, transform_passes=[ I64toI32(self._edge_compile_config._skip_dim_order), ], diff --git a/exir/capture/_config.py b/exir/capture/_config.py index eb699c53d9..a424757983 100644 --- a/exir/capture/_config.py +++ b/exir/capture/_config.py @@ -45,8 +45,7 @@ class EdgeCompileConfig: ) _skip_type_promotion: bool = False # TODO(gasoonjia): remove this - # TODO(T192537614): reenanle dim order as default - _skip_dim_order: bool = True + _skip_dim_order: bool = False @compatibility(is_backward_compatible=False)