Function require_torch_accelerator_with_fp64

tests/testing_utils.py:592–596

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

(test_case)

def require_torch_accelerator_with_fp64(test_case):
    """Decorator marking a test that requires an accelerator with support for the FP64 data type."""
    return pytest.mark.skipif(
        not _is_torch_fp64_available(torch_device), reason="test requires accelerator with fp64 support"
    )(test_case)


def require_big_gpu_with_torch_cuda(test_case):