6
6
7
7
from typing import List , Tuple
8
8
9
+ import pytest
9
10
import torch
10
11
from executorch .backends .arm .test import common
11
12
from executorch .backends .arm .test .tester .test_pipeline import (
@@ -117,25 +118,6 @@ def forward(self, x: torch.Tensor):
117
118
}
118
119
119
120
120
- fvp_xfails = {
121
- "any_rank1" : "MLETORCH-706 Support ScalarType::Bool in EthosUBackend." ,
122
- "any_rank1_squeeze" : "MLETORCH-706: Support ScalarType::Bool in EthosUBackend." ,
123
- "any_rank2" : "MLETORCH-706: Support ScalarType::Bool in EthosUBackend." ,
124
- "any_rank2_squeeze" : "MLETORCH-706: Support ScalarType::Bool in EthosUBackend." ,
125
- "any_rank2_dims" : "MLETORCH-706: Support ScalarType::Bool in EthosUBackend." ,
126
- "any_rank2_dims_squeeze" : "MLETORCH-706: Support ScalarType::Bool in EthosUBackend." ,
127
- "any_rank3_dims_squeeze" : "MLETORCH-706: Support ScalarType::Bool in EthosUBackend." ,
128
- "any_rank4" : "MLETORCH-706: Support ScalarType::Bool in EthosUBackend." ,
129
- "any_rank4_squeeze" : "MLETORCH-706: Support ScalarType::Bool in EthosUBackend." ,
130
- "any_rank4_dims" : "MLETORCH-706: Support ScalarType::Bool in EthosUBackend." ,
131
- "any_rank4_dims_squeeze" : "MLETORCH-706: Support ScalarType::Bool in EthosUBackend." ,
132
- "any_rank1_reduce_all" : "MLETORCH-706: Support ScalarType::Bool in EthosUBackend." ,
133
- "any_rank2_reduce_all" : "MLETORCH-706: Support ScalarType::Bool in EthosUBackend." ,
134
- "any_rank3_reduce_all" : "MLETORCH-706: Support ScalarType::Bool in EthosUBackend." ,
135
- "any_rank4_reduce_all" : "MLETORCH-706: Support ScalarType::Bool in EthosUBackend." ,
136
- }
137
-
138
-
139
121
@common .parametrize ("test_data" , test_data )
140
122
def test_any_tosa_MI (test_data : input_t1 ):
141
123
op , test_input = test_data
@@ -147,13 +129,13 @@ def test_any_tosa_MI(test_data: input_t1):
147
129
def test_any_tosa_BI (test_data : input_t1 ):
148
130
op , test_input = test_data
149
131
pipeline = TosaPipelineBI [input_t1 ](op , test_input , op .aten_op , op .exir_op )
150
- pipeline .pop_stage (pipeline .find_pos ("quantize" ) + 1 )
151
132
pipeline .pop_stage ("quantize" )
133
+ pipeline .pop_stage ("check.quant_nodes" )
152
134
pipeline .run ()
153
135
154
136
155
137
@common .parametrize ("test_data" , test_data )
156
- def test_logical_u55_BI (test_data : input_t1 ):
138
+ def test_any_u55_BI (test_data : input_t1 ):
157
139
# Tests that we don't delegate these ops since they are not supported on U55.
158
140
op , test_input = test_data
159
141
pipeline = OpNotSupportedPipeline [input_t1 ](
@@ -163,23 +145,13 @@ def test_logical_u55_BI(test_data: input_t1):
163
145
164
146
165
147
@common .parametrize ("test_data" , test_data )
166
- def test_floor_u85_BI (test_data : input_t1 ):
167
- op , test_input = test_data
168
- pipeline = EthosU85PipelineBI [input_t1 ](
169
- op , test_input , op .aten_op , op .exir_op , run_on_fvp = False
170
- )
171
- pipeline .pop_stage (pipeline .find_pos ("quantize" ) + 1 )
172
- pipeline .pop_stage ("quantize" )
173
- pipeline .run ()
174
-
175
-
176
- @common .parametrize ("test_data" , test_data , fvp_xfails )
177
- @common .SkipIfNoCorstone320
178
- def test_floor_u85_BI_on_fvp (test_data : input_t1 ):
148
+ @pytest .mark .xfail (reason = "MLETORCH-706: Support ScalarType::Bool in EthosUBackend." )
149
+ @common .XfailIfNoCorstone320
150
+ def test_any_u85_BI (test_data : input_t1 ):
179
151
op , test_input = test_data
180
152
pipeline = EthosU85PipelineBI [input_t1 ](
181
153
op , test_input , op .aten_op , op .exir_op , run_on_fvp = True
182
154
)
183
- pipeline .pop_stage (pipeline .find_pos ("quantize" ) + 1 )
184
155
pipeline .pop_stage ("quantize" )
156
+ pipeline .pop_stage ("check.quant_nodes" )
185
157
pipeline .run ()
0 commit comments