Skip to content

Commit bd1c979

Browse files
Martin LindströmMartin Lindström
authored andcommitted
Arm backend: Reenable test_fuse_const_ops_tosa_BI
The fusing of constant ops now works for TOSA BI again. Change-Id: I7f6d24dfc56b2cc3f5de6107c143be4f249c720b
1 parent 7993bb2 commit bd1c979

File tree

1 file changed

+8
-8
lines changed

1 file changed

+8
-8
lines changed

backends/arm/test/passes/test_fuse_constant_ops_pass.py

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,6 @@
44
# LICENSE file in the root directory of this source tree.
55

66
import operator
7-
import unittest
87
from typing import Tuple
98

109
import torch
@@ -13,10 +12,7 @@
1312
FuseConstantArgsPass,
1413
)
1514
from executorch.backends.arm.test import common
16-
from executorch.backends.arm.test.tester.test_pipeline import (
17-
PassPipeline,
18-
TosaPipelineBI,
19-
)
15+
from executorch.backends.arm.test.tester.test_pipeline import PassPipeline
2016

2117
input_t = Tuple[torch.Tensor] # Input x
2218

@@ -111,10 +107,14 @@ def test_fuse_const_ops_tosa_MI(module):
111107
pipeline.run()
112108

113109

114-
@unittest.skip("Test failing on internal CI")
115110
@common.parametrize("module", modules)
116111
def test_fuse_const_ops_tosa_BI(module):
117-
pipeline = TosaPipelineBI[input_t](
118-
module, (torch.rand(10, 10),), [], [], use_to_edge_transform_and_lower=True
112+
pipeline = PassPipeline[input_t](
113+
module,
114+
(torch.rand(10, 10),),
115+
tosa_version="TOSA-0.80+BI",
116+
ops_before_pass=module.ops_before_pass,
117+
ops_after_pass=module.ops_after_pass,
118+
passes_with_exported_program=[ComputeConstantOpsAOT, FuseConstantArgsPass],
119119
)
120120
pipeline.run()

0 commit comments

Comments
 (0)