File tree Expand file tree Collapse file tree 2 files changed +2
-10
lines changed Expand file tree Collapse file tree 2 files changed +2
-10
lines changed Original file line number Diff line number Diff line change @@ -430,16 +430,8 @@ def test_convolution_2d_tosa_MI_avgpool2d(test_data: torch.Tensor):
430
430
pipeline .run ()
431
431
432
432
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
-
441
433
@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 )
443
435
def test_convolution_2d_tosa_BI_avgpool2d (test_data : torch .Tensor ):
444
436
model = ComboConvAvgPool2d ()
445
437
pipeline = TosaPipelineBI [input_t1 ](
Original file line number Diff line number Diff line change @@ -282,7 +282,7 @@ def __init__(
282
282
custom_path : str = None ,
283
283
atol : float = 1e-03 ,
284
284
rtol : float = 1e-03 ,
285
- qtol : int = 0 ,
285
+ qtol : int = 1 ,
286
286
):
287
287
tosa_profiles = {
288
288
"0.80" : TosaSpecification .create_from_string ("TOSA-0.80+BI" ),
You can’t perform that action at this time.
0 commit comments