We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 274c9aa commit 72b36e1Copy full SHA for 72b36e1
tests/accelerators/test_gpu.py
@@ -63,17 +63,17 @@ def test_set_cuda_device(set_device_mock, tmpdir):
63
64
65
@RunIf(min_gpus=1)
66
-def test_gpu_availability():
+def test_real_gpu_availability():
67
assert GPUAccelerator.is_available()
68
69
70
@mock.patch("torch.cuda.is_available", return_value=True)
71
@mock.patch("torch.cuda.device_count", return_value=2)
72
-def test_mocked_gpu_available(*_):
+def test_gpu_available(*_):
73
74
75
76
@mock.patch("torch.cuda.is_available", return_value=False)
77
@mock.patch("torch.cuda.device_count", return_value=0)
78
-def test_mocked_gpu_availability(*_):
+def test_gpu_not_available(*_):
79
assert not GPUAccelerator.is_available()
0 commit comments