Class GPUInfo

src/whichllm/hardware/types.py:7–17

@dataclass
class GPUInfo:
    name: str
    vendor: str  # "nvidia" | "amd" | "apple" | "intel"
    vram_bytes: int
    usable_vram_bytes: int | None = None
    compute_capability: tuple[int, int] | None = None  # NVIDIA only
    cuda_version: str | None = None
    rocm_version: str | None = None
    memory_bandwidth_gbps: float | None = None  # from lookup table
    shared_memory: bool = False
    vram_overridden: bool = False


@dataclass