Skip to content

GitLab

Commit 199be867 authored by rusty1s's avatar rusty1s
Browse files

skip if cuda not available

parent 1f2c3c31
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
test/test_fps.py
View file @ 199be867
......@@ -9,6 +9,7 @@ from .utils import tensor, grad_dtypes
devices = [torch.device('cuda')]
@pytest.mark.skipif(not torch.cuda.is_available(), reason='CUDA not available')
@pytest.mark.parametrize('dtype,device', product(grad_dtypes, devices))
def test_fps(dtype, device):
x = tensor([
......@@ -27,6 +28,7 @@ def test_fps(dtype, device):
assert out.tolist() == [0, 2, 4, 6]
@pytest.mark.skipif(not torch.cuda.is_available(), reason='CUDA not available')
@pytest.mark.parametrize('dtype,device', product(grad_dtypes, devices))
def test_fps_speed(dtype, device):
return
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment