From fe9120f43e6e1f2c44af895303a3ef8f77f55b89 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Martin=20Lindstr=C3=B6m?= Date: Mon, 5 May 2025 16:09:21 +0200 Subject: [PATCH] Arm backend: Reenable test_fuse_const_ops_tosa_BI The fusing of constant ops now works for TOSA BI again. Change-Id: I7f6d24dfc56b2cc3f5de6107c143be4f249c720b --- .../test/passes/test_fuse_constant_ops_pass.py | 15 +++++---------- 1 file changed, 5 insertions(+), 10 deletions(-) diff --git a/backends/arm/test/passes/test_fuse_constant_ops_pass.py b/backends/arm/test/passes/test_fuse_constant_ops_pass.py index 5e759d7a824..4ec6942430f 100644 --- a/backends/arm/test/passes/test_fuse_constant_ops_pass.py +++ b/backends/arm/test/passes/test_fuse_constant_ops_pass.py @@ -4,7 +4,6 @@ # LICENSE file in the root directory of this source tree. import operator -import unittest from typing import Tuple import torch @@ -13,10 +12,7 @@ FuseConstantArgsPass, ) from executorch.backends.arm.test import common -from executorch.backends.arm.test.tester.test_pipeline import ( - PassPipeline, - TosaPipelineBI, -) +from executorch.backends.arm.test.tester.test_pipeline import PassPipeline input_t = Tuple[torch.Tensor] # Input x @@ -111,15 +107,14 @@ def test_fuse_const_ops_tosa_MI(module: torch.nn.Module): pipeline.run() -@unittest.skip("Test failing on internal CI") @common.parametrize("module", modules) def test_fuse_const_ops_tosa_BI(module: torch.nn.Module): - pipeline = TosaPipelineBI[input_t]( + pipeline = PassPipeline[input_t]( module, (torch.rand(10, 10),), - [], - [], quantize=True, - use_to_edge_transform_and_lower=True, + ops_before_pass=module.ops_before_pass, + ops_after_pass=module.ops_after_pass, + passes_with_exported_program=[ComputeConstantOpsAOT, FuseConstantArgsPass], ) pipeline.run()