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):
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 )
443435def test_convolution_2d_tosa_BI_avgpool2d (test_data : torch .Tensor ):
444436 model = ComboConvAvgPool2d ()
445437 pipeline = TosaPipelineBI [input_t1 ](
Original file line number Diff line number Diff 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" ),
You can’t perform that action at this time.
0 commit comments