def require_torch_bf16_gpu(test_case):
    """Decorator marking a test that requires torch>=1.10, using Ampere GPU or newer arch with cuda>=11.0"""
    return unittest.skipUnless(
        is_torch_bf16_gpu_available(),
        "test requires torch>=1.10, using Ampere GPU or newer arch with cuda>=11.0",
    )(test_case)
Decorator marking a test that requires torch>=1.10, using Ampere GPU or newer arch with cuda>=11.0
dpkd/transformers/src/transformers/testing_utils.py