diff --git a/tests/python/topi/python/test_topi_argwhere.py b/tests/python/topi/python/test_topi_argwhere.py index 5cb7cd44513e..433030863a43 100644 --- 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)