Skip to content

Commit 94d1381

Browse files
martinlsmMartin Lindström
andauthored
Arm backend: Reenable test_fuse_const_ops_tosa_BI (#10847)
The fusing of constant ops now works for TOSA BI again. Co-authored-by: Martin Lindström <[email protected]>
1 parent 54e7c75 commit 94d1381

File tree

1 file changed

+5
-10
lines changed

1 file changed

+5
-10
lines changed

backends/arm/test/passes/test_fuse_constant_ops_pass.py

Lines changed: 5 additions & 10 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,15 +107,14 @@ def test_fuse_const_ops_tosa_MI(module: torch.nn.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: torch.nn.Module):
117-
pipeline = TosaPipelineBI[input_t](
112+
pipeline = PassPipeline[input_t](
118113
module,
119114
(torch.rand(10, 10),),
120-
[],
121-
[],
122115
quantize=True,
123-
use_to_edge_transform_and_lower=True,
116+
ops_before_pass=module.ops_before_pass,
117+
ops_after_pass=module.ops_after_pass,
118+
passes_with_exported_program=[ComputeConstantOpsAOT, FuseConstantArgsPass],
124119
)
125120
pipeline.run()

0 commit comments

Comments
 (0)