From 7173c39bac9f2a0f503ea8e80f3d1383ed8b1af3 Mon Sep 17 00:00:00 2001 From: Eric Lunderberg Date: Tue, 17 May 2022 15:22:24 -0500 Subject: [PATCH] Corrected mark from requires_gpu to requires_cuda --- tests/python/integration/test_reduce.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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")