[ManifestAlloc] Handle TupleType inputs in CheckReshapeOnly

apache · Oct 28, 2020 · 38420a6 · 38420a6
1 parent 8a23866
commit 38420a6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/python/relay/test_vm.py b/tests/python/relay/test_vm.py
@@ -766,7 +766,7 @@ def test_vm_reshape_tuple(x_shape=(1, 4, 2), y_shape=(1, 2, 10)):
     y_data = np.random.uniform(size=y_shape).astype("float32")
 
     for tgt, ctx in tvm.testing.enabled_targets():
-        res = veval(f, (x_data, y_data))
+        res = veval(f, (x_data, y_data), ctx=tvm.cpu(), target="llvm")
         tvm.testing.assert_allclose(res.asnumpy(), np.reshape(x_data, (1, -1)))