Corrected mark from requires_gpu to requires_cuda

apache · May 17, 2022 · 7173c39 · 7173c39
1 parent fb9ec20
commit 7173c39
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/python/integration/test_reduce.py b/tests/python/integration/test_reduce.py
@@ -528,7 +528,7 @@ def check_target(device):
     check_target("rocm")
 
 
-@tvm.testing.requires_gpu
+@tvm.testing.requires_cuda
 def test_reduce_storage_reuse():
     target = tvm.target.Target("cuda")