change requires_gpu to requires_cuda

apache · Jan 3, 2024 · fcbd65d · fcbd65d
1 parent 82d7724
commit fcbd65d
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/tests/python/contrib/test_msc/test_runner.py b/tests/python/contrib/test_msc/test_runner.py
@@ -109,8 +109,8 @@ def test_tvm_runner_cpu():
     _test_from_torch(TVMRunner, "cpu", is_training=True)
 
 
-@tvm.testing.requires_gpu
-def test_tvm_runner_gpu():
+@tvm.testing.requires_cuda
+def test_tvm_runner_cuda():
     """Test runner for tvm on cuda"""
 
     _test_from_torch(TVMRunner, "cuda", is_training=True)
@@ -122,8 +122,8 @@ def test_torch_runner_cpu():
     _test_from_torch(TorchRunner, "cpu")
 
 
-@tvm.testing.requires_gpu
-def test_torch_runner_gpu():
+@tvm.testing.requires_cuda
+def test_torch_runner_cuda():
     """Test runner for torch on cuda"""
 
     _test_from_torch(TorchRunner, "cuda", atol=1e-1, rtol=1e-1)

diff --git a/tests/python/contrib/test_msc/test_tools.py b/tests/python/contrib/test_msc/test_tools.py
@@ -251,7 +251,7 @@ def test_tvm_tool(tool_type):
     )
 
 
-@tvm.testing.requires_gpu
+@tvm.testing.requires_cuda
 @pytest.mark.parametrize("tool_type", [ToolType.PRUNER, ToolType.QUANTIZER])
 def test_tvm_distill(tool_type):
     """Test tools for tvm with distiller"""