Skip to content

GitLab

Commit 19d744fd authored by rusty1s's avatar rusty1s
Browse files

gpu test bugfix

parent 20a6cc50
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
test/test_spline_conv.py
View file @ 19d744fd
...@@ -93,6 +93,7 @@ def test_spline_conv_gpu(tensor): ...@@ -93,6 +93,7 @@ def test_spline_conv_gpu(tensor):
assert output.cpu().tolist() == expected_output.tolist() assert output.cpu().tolist() == expected_output.tolist()
@pytest.mark.skipif(not torch.cuda.is_available(), reason='no CUDA')
def test_spline_weighting_backward_gpu(): def test_spline_weighting_backward_gpu():
for degree in implemented_degrees.keys(): for degree in implemented_degrees.keys():
kernel_size = torch.cuda.LongTensor([5, 5, 5]) kernel_size = torch.cuda.LongTensor([5, 5, 5])
......
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