Skip to content

T
tic
Commit ccd33332 by masahi Committed by Tianqi Chen
Options

Compare relay and numpy outputs in graph runtime test (#2164)

parent 9473dca2
Showing with 1 additions and 0 deletions
tests/python/relay/test_backend_graph_runtime.py
......@@ -30,6 +30,7 @@ def check_rts(expr, args, expected_result, mod=None):
eval_result = intrp.evaluate(expr)(*args)
rts_result = graph.evaluate(expr)(*args)
tvm.testing.assert_allclose(eval_result.asnumpy(), rts_result.asnumpy())
tvm.testing.assert_allclose(eval_result.asnumpy(), expected_result)
def test_add_op_scalar():
"""
......
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