From ac6ba45272b907f79057e0eb4368a898fe18ca22 Mon Sep 17 00:00:00 2001 From: "Tugsbayasgalan (Tugsuu) Manlaibaatar" Date: Fri, 25 Oct 2024 10:22:29 -0700 Subject: [PATCH] All executprch test pass now Summary: Title Reviewed By: tarun292 Differential Revision: D64918434 --- exir/tests/test_memory_format_ops_pass_utils.py | 2 +- exir/tests/test_passes.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/exir/tests/test_memory_format_ops_pass_utils.py b/exir/tests/test_memory_format_ops_pass_utils.py index dc02d257386..8ae5c0190a4 100644 --- a/exir/tests/test_memory_format_ops_pass_utils.py +++ b/exir/tests/test_memory_format_ops_pass_utils.py @@ -69,7 +69,7 @@ class MemoryFormatOpsPassTestUtils: def memory_format_test_runner( test_class: unittest.TestCase, test_set: MemoryFormatTestSet ): - before = export(test_set.module, test_set.sample_input) + before = export(test_set.module, test_set.sample_input).run_decompositions({}) if test_set.use_xnnpack: epm = to_edge_transform_and_lower( diff --git a/exir/tests/test_passes.py b/exir/tests/test_passes.py index a3aa1f39975..34aedd51d06 100644 --- a/exir/tests/test_passes.py +++ b/exir/tests/test_passes.py @@ -1735,7 +1735,7 @@ def _do_checks( ep = export( m, (input,), - ) + ).run_decompositions({}) _do_checks( ep.graph_module.code, aten_op_str,