We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 182c64e commit b0e048dCopy full SHA for b0e048d
tests/python/relay/test_op_level1.py
@@ -527,15 +527,17 @@ def test_concatenate3(target, dev):
527
t_shape = [3, 2, 2, ending]
528
do_concat_test(shapes, t_shape, dtype, axis, dev, target)
529
530
+
531
@tvm.testing.parametrize_targets("llvm")
532
def test_concatenate4(target, dev):
533
np.random.seed(7)
534
x_shape = (2, 1)
- x = relay.var("x", shape=x_shape, dtype='int64')
535
+ x = relay.var("x", shape=x_shape, dtype="int64")
536
concat = relay.concatenate([x], axis=1)
537
f = relay.Function([x], concat)
- x_val = np.array([[33], [13]], dtype='int64')
538
- op_res = relay.create_executor('graph', device=tvm.cpu(), target='llvm').evaluate(f)(x_val)
+ x_val = np.array([[33], [13]], dtype="int64")
539
+ graph = relay.create_executor("graph", device=tvm.cpu(), target="llvm")
540
+ op_res = graph.evaluate(f)(x_val)
541
ref_res = np.concatenate([x_val], axis=1)
542
tvm.testing.assert_allclose(op_res.numpy(), ref_res, rtol=0.000001)
543
0 commit comments