Function require_torch_accelerator_with_fp16

tests/testing_utils.py:585–589

Decorator marking a test that requires an accelerator with support for the FP16 data type.

(test_case)

def require_torch_accelerator_with_fp16(test_case):
    """Decorator marking a test that requires an accelerator with support for the FP16 data type."""
    return pytest.mark.skipif(
        not _is_torch_fp16_available(torch_device), reason="test requires accelerator with fp16 support"
    )(test_case)


def require_torch_accelerator_with_fp64(test_case):