From ade8d900eeaa77526112b7b9e64dc003b64ec93d Mon Sep 17 00:00:00 2001 From: Eric Lunderberg Date: Wed, 2 Jun 2021 11:18:13 -0700 Subject: [PATCH] [Vulkan][UnitTests] Compatibility fix for test_vulkan_unique(). relay.unique return values changed in 6baccc13, updating vulkan unit tests to match. --- tests/python/unittest/test_target_codegen_vulkan.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/python/unittest/test_target_codegen_vulkan.py b/tests/python/unittest/test_target_codegen_vulkan.py index 2770ae5878d0..dc165331729e 100644 --- a/tests/python/unittest/test_target_codegen_vulkan.py +++ b/tests/python/unittest/test_target_codegen_vulkan.py @@ -255,7 +255,7 @@ def test_vulkan_unique(): dtype = "int32" x = relay.var("x", shape=(relay.Any(),), dtype=dtype) mod = tvm.IRModule() - [unique, _, num_unique] = relay.unique(x, is_sorted=True) + [unique, _, _, num_unique] = relay.unique(x, is_sorted=True) mod["main"] = relay.Function([x], relay.op.strided_slice(unique, begin=[0], end=num_unique)) x_np = np.random.randint(0, high=10, size=(10,)).astype(dtype) res_np = np.unique(x_np)