Function _sm_version

src/fused_mm_sampling/cuda_impl.py:35–38

Return the SM version string (e.g. '86' for RTX 3090).

()

def _sm_version() -> str:
    """Return the SM version string (e.g. '86' for RTX 3090)."""
    major, minor = torch.cuda.get_device_capability()
    return f"{major}{minor}"


TILE_V = 128