Function require_torch_gpu

diffusers/src/diffusers/utils/testing_utils.py:267–271

Decorator marking a test that requires CUDA and PyTorch.

(test_case)

def require_torch_gpu(test_case):
    """Decorator marking a test that requires CUDA and PyTorch."""
    return unittest.skipUnless(is_torch_available() and torch_device == "cuda", "test requires PyTorch+CUDA")(
        test_case
    )


# These decorators are for accelerator-specific behaviours that are not GPU-specific