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 44# LICENSE file in the root directory of this source tree.
55
66import operator
7- import unittest
87from typing import Tuple
98
109import torch
1312 FuseConstantArgsPass ,
1413)
1514from 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
2117input_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 )
116111def 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 ()
You can’t perform that action at this time.
0 commit comments