File tree Expand file tree Collapse file tree 1 file changed +2
-6
lines changed Expand file tree Collapse file tree 1 file changed +2
-6
lines changed Original file line number Diff line number Diff line change @@ -333,9 +333,7 @@ def decorator(test_case):
333333
334334def require_torch_gpu (test_case ):
335335 """Decorator marking a test that requires CUDA and PyTorch."""
336- return pytest .mark .skipif (
337- torch_device != "cuda" , reason = "test requires PyTorch+CUDA"
338- )(test_case )
336+ return pytest .mark .skipif (torch_device != "cuda" , reason = "test requires PyTorch+CUDA" )(test_case )
339337
340338
341339def require_torch_cuda_compatibility (expected_compute_capability ):
@@ -354,9 +352,7 @@ def decorator(test_case):
354352# These decorators are for accelerator-specific behaviours that are not GPU-specific
355353def require_torch_accelerator (test_case ):
356354 """Decorator marking a test that requires an accelerator backend and PyTorch."""
357- return pytest .mark .skipif (
358- torch_device == "cpu" , reason = "test requires accelerator+PyTorch"
359- )(test_case )
355+ return pytest .mark .skipif (torch_device == "cpu" , reason = "test requires accelerator+PyTorch" )(test_case )
360356
361357
362358def require_torch_multi_gpu (test_case ):
You can’t perform that action at this time.
0 commit comments