From 2d8caf86d155a395333a823db6b96147bcad8d9c Mon Sep 17 00:00:00 2001 From: Andrew Luo Date: Mon, 20 Sep 2021 14:44:34 -0700 Subject: [PATCH] bump tolerances --- tests/python/frontend/onnx/test_forward.py | 11 ++++++++++- 1 file changed, 10 insertions(+), 1 deletion(-) diff --git a/tests/python/frontend/onnx/test_forward.py b/tests/python/frontend/onnx/test_forward.py index 91d3911da530..664ed360ceba 100644 --- 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",