[FIX] disable cuda test for argwhere (apache#7042)

* disable cuda test for argwhere * Fix lint Co-authored-by: Lianmin Zheng <lianminzheng@gmail.com>
electriclilies · Feb 18, 2021 · 7ddb9f7 · 7ddb9f7
1 parent 3b7d726
commit 7ddb9f7
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/tests/python/topi/python/test_topi_argwhere.py b/tests/python/topi/python/test_topi_argwhere.py
@@ -60,6 +60,9 @@ def check_device(device, ctx):
         tvm.testing.assert_allclose(args[-1].asnumpy(), np.array(np_out))
 
     for target, ctx in tvm.testing.enabled_targets():
+        # TODO(zhiics) Enable argwhere gpu test after sort is fixed.
+        if ctx.device_type != 1:
+            continue
         check_device(target, ctx)