def require_torch_tf32(test_case):
    """Decorator marking a test that requires Ampere or a newer GPU arch, cuda>=11 and torch>=1.7."""
    return unittest.skipUnless(
        is_torch_tf32_available(), "test requires Ampere or a newer GPU arch, cuda>=11 and torch>=1.7"
    )(test_case)
Decorator marking a test that requires Ampere or a newer GPU arch, cuda>=11 and torch>=1.7.
dpkd/transformers/src/transformers/testing_utils.py