Skip to content

Commit 1e337c2

Browse files
committed
try disable new vk test to see if CI error still repros
1 parent 80a7b52 commit 1e337c2

File tree

2 files changed

+2
-8
lines changed

2 files changed

+2
-8
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", "vulkan -from_device=0"]:
131+
for device in ["cuda"]:
132132
check_device(device)
133133

134134

tests/python/topi/python/test_topi_conv2d_int8.py

Lines changed: 1 addition & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -371,12 +371,6 @@ def get_ref_data():
371371
topi.cuda.schedule_conv2d_NCHWc_int8,
372372
4,
373373
),
374-
(
375-
"vulkan -from_device=0",
376-
lambda a, w, s, p, d, l, ol, o: topi.cuda.conv2d_NCHWc_int8(a, w, s, p, d, l, o),
377-
topi.cuda.schedule_conv2d_NCHWc_int8,
378-
4,
379-
),
380374
(
381375
"llvm -device arm_cpu -mtriple aarch64-linux-gnu -mattr=+neon",
382376
topi.arm_cpu.conv2d_NCHWc_int8,
@@ -506,7 +500,7 @@ def check_target(target):
506500

507501
verify_workload_padding()
508502

509-
for target in ["cuda", "vulkan -from_device=0"]:
503+
for target in ["cuda"]:
510504
check_target(target)
511505

512506

0 commit comments

Comments
 (0)