enable more flaky tests

Author: mbrookhart

tests/python/relay/test_any.py

@@ -837,8 +837,7 @@ def verify_any_topk(data_shape, kval, np_dshape, dtype, const_k=False):
     check_result(in_vals, mod, ref_out)
 
 
-# TODO(kevinthesun): enable this test when Thrust is available in ci.
-# @tvm.testing.uses_gpu
+@tvm.testing.uses_gpu
 def test_any_topk():
     verify_any_topk(any_dims(1), 5, (10,), "float32")
     verify_any_topk(any_dims(2), 2, (6, 3), "int32")
@@ -1387,7 +1386,6 @@ def test_any_where():
         any_dims(2), any_dims(2), any_dims(2), (3, 4), (3, 1), (1, 4), y_np_shape_invalid=(2, 4)
     )
 
-
 # TODO(kevinthesun): enable gpu test when Thrust is available in ci.
 # @tvm.testing.uses_gpu
 def test_non_max_suppression():

tests/python/topi/python/test_topi_argwhere.py

@@ -66,9 +66,7 @@ def check_device(device, ctx):
         check_device(target, ctx)
 
 
-# TODO(zhiics) Enable argwhere gpu test after sort is fixed. Otherwise, we have
-# to use thrust to guarantee the correct results which has been tested locally.
-# @tvm.testing.uses_gpu
+@tvm.testing.uses_gpu
 def test_argwhere():
     verify_argwhere((1,))
     verify_argwhere((100,))