Function require_torch_accelerator_with_training

tests/testing_utils.py:648–653

Decorator marking a test that requires an accelerator with support for training.

(test_case)

def require_torch_accelerator_with_training(test_case):
    """Decorator marking a test that requires an accelerator with support for training."""
    return pytest.mark.skipif(
        not (is_torch_available() and backend_supports_training(torch_device)),
        reason="test requires accelerator with training support",
    )(test_case)


def skip_mps(test_case):