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 3ceac24 commit 635d485Copy full SHA for 635d485
tests/testing_utils.py
@@ -334,7 +334,7 @@ def decorator(test_case):
334
def require_torch_gpu(test_case):
335
"""Decorator marking a test that requires CUDA and PyTorch."""
336
return pytest.mark.skipif(
337
- not (is_torch_available() and torch_device == "cuda"), reason="test requires PyTorch+CUDA"
+ torch_device != "cuda", reason="test requires PyTorch+CUDA"
338
)(test_case)
339
340
0 commit comments