Skip to content

Commit 0a6f622

Browse files
authored
Arm Backend: Updated TosaPipelineBI default qtol (#10907)
### Summary * Updates the default qtol=1 * Removes x_fails from test_conv_combos
1 parent 43ab323 commit 0a6f622

File tree

2 files changed

+2
-10
lines changed

2 files changed

+2
-10
lines changed

backends/arm/test/ops/test_conv_combos.py

Lines changed: 1 addition & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -430,16 +430,8 @@ def test_convolution_2d_tosa_MI_avgpool2d(test_data: torch.Tensor):
430430
pipeline.run()
431431

432432

433-
x_fails = {
434-
"combo_conv_avgpool_20_x_4d": "AssertionError: Output 0 does not match reference output.",
435-
"combo_conv_avgpool_4d": "AssertionError: Output 0 does not match reference output.",
436-
"combo_conv_avgpool_5_x_4d_randn": "AssertionError: Output 0 does not match reference output.",
437-
"combo_conv_avgpool_2_x_4d": "AssertionError: Output 0 does not match reference output.",
438-
}
439-
440-
441433
@pytest.mark.flaky(reruns=5) # TODO: Investigate flakyness (MLTORCH-307)
442-
@common.parametrize("test_data", ComboConvAvgPool2d.test_data, x_fails)
434+
@common.parametrize("test_data", ComboConvAvgPool2d.test_data)
443435
def test_convolution_2d_tosa_BI_avgpool2d(test_data: torch.Tensor):
444436
model = ComboConvAvgPool2d()
445437
pipeline = TosaPipelineBI[input_t1](

backends/arm/test/tester/test_pipeline.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -282,7 +282,7 @@ def __init__(
282282
custom_path: str = None,
283283
atol: float = 1e-03,
284284
rtol: float = 1e-03,
285-
qtol: int = 0,
285+
qtol: int = 1,
286286
):
287287
tosa_profiles = {
288288
"0.80": TosaSpecification.create_from_string("TOSA-0.80+BI"),

0 commit comments

Comments
 (0)