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

FrozenGene · Nov 25, 2018 · 0fff6b8 · 0fff6b8
1 parent fd330f1
commit 0fff6b8
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/python/relay/test_backend_graph_runtime.py b/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():
     """