Skip to content

Commit c09ba06

Browse files
Arm backend: Reduce atol in vector_norm test (#11281)
pipeline.change_args("run_method_and_compare_outputs",...) was used to change qtol, but as no inputs were passed in the call run_method_and_compare_outputs produced new inputs with a different distribution (randn), than what was used for calibration (rand). Signed-off-by: Oscar Andersson <[email protected]>
1 parent 0a09b49 commit c09ba06

File tree

1 file changed

+1
-5
lines changed

1 file changed

+1
-5
lines changed

backends/arm/test/ops/test_linalg_vector_norm.py

Lines changed: 1 addition & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -72,7 +72,6 @@ def test_vector_norm_tosa_MI(test_module):
7272

7373
pipeline = TosaPipelineMI[input_t](model, input_tensor, aten_op, exir_op)
7474

75-
pipeline.change_args("run_method_and_compare_outputs", qtol=1, atol=1e-4, rtol=1e-4)
7675
pipeline.run()
7776

7877

@@ -90,7 +89,6 @@ def test_vector_norm_tosa_BI(test_module):
9089
exir_op,
9190
symmetric_io_quantization=True,
9291
)
93-
pipeline.change_args("run_method_and_compare_outputs", qtol=1, atol=1, rtol=1)
9492
pipeline.run()
9593

9694

@@ -107,13 +105,12 @@ def test_vector_norm_u55_BI_fvp(test_module):
107105
run_on_fvp=True,
108106
symmetric_io_quantization=True,
109107
)
110-
pipeline.change_args("run_method_and_compare_outputs", qtol=1, atol=1, rtol=1)
111108
pipeline.pop_stage("check_not.exir")
112109
pipeline.run()
113110

114111

115112
@common.parametrize("test_module", test_modules)
116-
@common.XfailIfNoCorstone300
113+
@common.XfailIfNoCorstone320
117114
def test_vector_norm_u85_BI_fvp(test_module):
118115
model, input_tensor = test_module
119116

@@ -126,6 +123,5 @@ def test_vector_norm_u85_BI_fvp(test_module):
126123
run_on_fvp=True,
127124
symmetric_io_quantization=True,
128125
)
129-
pipeline.change_args("run_method_and_compare_outputs", qtol=1, atol=1, rtol=1)
130126
pipeline.pop_stage("check_not.exir")
131127
pipeline.run()

0 commit comments

Comments
 (0)