File tree Expand file tree Collapse file tree 1 file changed +5
-10
lines changed Expand file tree Collapse file tree 1 file changed +5
-10
lines changed Original file line number Diff line number Diff line change 4
4
# LICENSE file in the root directory of this source tree.
5
5
6
6
import operator
7
- import unittest
8
7
from typing import Tuple
9
8
10
9
import torch
13
12
FuseConstantArgsPass ,
14
13
)
15
14
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
20
16
21
17
input_t = Tuple [torch .Tensor ] # Input x
22
18
@@ -111,15 +107,14 @@ def test_fuse_const_ops_tosa_MI(module: torch.nn.Module):
111
107
pipeline .run ()
112
108
113
109
114
- @unittest .skip ("Test failing on internal CI" )
115
110
@common .parametrize ("module" , modules )
116
111
def test_fuse_const_ops_tosa_BI (module : torch .nn .Module ):
117
- pipeline = TosaPipelineBI [input_t ](
112
+ pipeline = PassPipeline [input_t ](
118
113
module ,
119
114
(torch .rand (10 , 10 ),),
120
- [],
121
- [],
122
115
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 ],
124
119
)
125
120
pipeline .run ()
You can’t perform that action at this time.
0 commit comments