bump tolerances (apache#9054)

ylc · Sep 29, 2021 · c8b2f86 · c8b2f86
1 parent 155fe10
commit c8b2f86
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/tests/python/frontend/onnx/test_forward.py b/tests/python/frontend/onnx/test_forward.py
@@ -1840,7 +1840,16 @@ def verify_reduce_func(func, data, axis, keepdims):
 
         model = helper.make_model(graph, producer_name="reduce_test")
 
-        verify_with_ort_with_inputs(model, [data], [outshape], opset=11, target=target, dev=dev)
+        verify_with_ort_with_inputs(
+            model,
+            [data],
+            [outshape],
+            opset=11,
+            target=target,
+            dev=dev,
+            rtol=1e-4,
+            atol=1e-4,
+        )
 
     funcs = [
         "ReduceMax",