Skip to content

T
tic
Commit a5784757 by Joshua Z. Zhang Committed by Tianqi Chen
Options

relax rtol atol for div/log operators (#2400)

parent f607d46c
Showing with 1 additions and 1 deletions
nnvm/tests/python/frontend/onnx/test_forward.py
...@@ -910,7 +910,7 @@ def test_single_ops(): ...@@ -910,7 +910,7 @@ def test_single_ops():
model = helper.make_model(graph, producer_name='_test') model = helper.make_model(graph, producer_name='_test')
for target, ctx in ctx_list(): for target, ctx in ctx_list():
tvm_out = get_tvm_output(model, [x], target, ctx) tvm_out = get_tvm_output(model, [x], target, ctx)
tvm.testing.assert_allclose(out_np, tvm_out) tvm.testing.assert_allclose(out_np, tvm_out, rtol=1e-5, atol=1e-5)
x = np.random.uniform(size=in_shape).astype(dtype) x = np.random.uniform(size=in_shape).astype(dtype)
verify_single_ops("Neg",x, -x) verify_single_ops("Neg",x, -x)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment