Skip to content

Commit cf00438

Browse files
committed
add back test_topi_batch_matmul.py change
1 parent 8d50235 commit cf00438

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/python/topi/python/test_topi_batch_matmul.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -128,7 +128,7 @@ def check_device(device):
128128
f(a, b, c)
129129
tvm.testing.assert_allclose(c.numpy(), c_np, rtol=1e-5)
130130

131-
for device in ["cuda"]:
131+
for device in ["cuda", "vulkan -from_device=0"]:
132132
check_device(device)
133133

134134

0 commit comments

Comments
 (0)