Function _get_max_cc

tests/transforms/test_spatial_gpu_support.py:24–32

Return the max compute capability (major*100+minor) the _C extension was compiled for, or 0 if the extension is not available (no build info).

()

def _get_max_cc() -> int:
    """Return the max compute capability (major*100+minor) the _C extension was compiled for,
    or 0 if the extension is not available (no build info)."""
    try:
        from monai._C import max_compute_capability

        return int(max_compute_capability())
    except (ImportError, AttributeError):
        return 0


def _has_sm120_support() -> bool: