diff --git a/tests/python/integration/test_reduce.py b/tests/python/integration/test_reduce.py index a40164ded941e..f3886374ccb65 100644 --- 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")