Function _lookup_dbgpu

src/whichllm/hardware/gpu_simulator.py:154–201

Look up GPU spec from dbgpu database. Returns GPUSpecification or None.

(name: str)

def _lookup_dbgpu(name: str) -> GPUSpecification | None:
    """Look up GPU spec from dbgpu database. Returns GPUSpecification or None."""
    from dbgpu import GPUDatabase

    db = GPUDatabase.default()

    # Normalize input: "GTX1080" → "GTX 1080"
    normalized = _normalize_gpu_name(name)
    compact = re.sub(r"\s+", "", normalized.lower())
    names_to_try = [name] if normalized == name else [name, normalized]
    alias_hits = _COMMON_GPU_ALIASES.get(compact)
    if alias_hits:
        names_to_try.extend(alias_hits)

    for n in names_to_try:
        # 1) Exact key lookup
        try:
            return db[n]
        except KeyError:
            pass

        # 2) Try with common manufacturer prefixes
        for prefix in _MANUFACTURER_PREFIXES:
            try:
                return db[prefix + n]
            except KeyError:
                pass

        # 3) Substring match with word-boundary filtering
        result = _substring_search(db, n)
        if result is not None:
            return result

    # 4) Fuzzy search as last resort (use normalized name + token_set_ratio)
    try:
        from thefuzz import fuzz, process

        results = process.extract(
            normalized, db.names, limit=3, scorer=fuzz.token_set_ratio
        )
        if results and results[0][1] >= 90:
            return db[results[0][0]]
        # Store top suggestions for error messages
        if results:
            _last_suggestions[:] = [(n, s) for n, s in results if s >= 70]
    except ImportError:
        pass
    return None


# Mutable list to pass suggestions from lookup to error message