Skip to content

Commit 6a54833

Browse files
committed
fix ut
1 parent ee2579a commit 6a54833

File tree

2 files changed

+5
-1
lines changed

2 files changed

+5
-1
lines changed

_unittests/ut_light_api/test_backend_export.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -242,7 +242,7 @@ def run_node(cls, node, inputs, device=None, outputs_info=None, **kwargs):
242242

243243
# The following tests are too slow with the reference implementation (Conv).
244244
backend_test.exclude(
245-
"(FLOAT8|BFLOAT16|INT4|_opt_|_3d_|_momentum_|_4d_"
245+
"(FLOAT8|BFLOAT16|INT4|_opt_|_3d_|_momentum_|_4d_|int4"
246246
"|test_adagrad"
247247
"|test_adam"
248248
"|test_ai_onnx_ml_"
@@ -270,6 +270,8 @@ def run_node(cls, node, inputs, device=None, outputs_info=None, **kwargs):
270270
"|test_squeezenet"
271271
"|test_vgg19"
272272
"|test_zfnet512"
273+
"|test_range_float_type_positive_delta_expanded"
274+
"|test_range_int32_type_negative_delta_expanded"
273275
")"
274276
)
275277

_unittests/ut_reference/test_backend_extended_reference_evaluator.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -164,8 +164,10 @@ def run_node(cls, node, inputs, device=None, outputs_info=None, **kwargs):
164164
"|test_cast_no_saturate_"
165165
"|_to_FLOAT8"
166166
"|_FLOAT8"
167+
"|INT4"
167168
"|test_quantizelinear_e4m3fn"
168169
"|test_quantizelinear_e5m2"
170+
"|test_scatter_with"
169171
")"
170172
)
171173

0 commit comments

Comments
 (0)