wehub-resource-sync
119 lines
4.4 KiB
Python
119 lines
4.4 KiB
Python
"""Tests for dbgpu-backed bandwidth resolution on detected hardware.
|
|
|
|
The detection path must never give a laptop/mobile card its desktop bandwidth
|
|
(issues #74, #61, #93). These tests pin the strict matching rules: curated
|
|
values win, the curated lookup is mobile-aware, dbgpu fills gaps, and an
|
|
unknown card resolves to None rather than a wrong guess.
|
|
"""
|
|
|
|
from whichllm.hardware.gpu_db import (
|
|
_normalize_detected_name,
|
|
_static_bandwidth,
|
|
resolve_detected_bandwidth,
|
|
)
|
|
|
|
_GiB = 1024**3
|
|
|
|
|
|
# --- normalization ---------------------------------------------------------
|
|
|
|
|
|
def test_normalize_strips_vendor_trademark_and_maps_laptop_to_mobile():
|
|
assert (
|
|
_normalize_detected_name("NVIDIA GeForce RTX 5090 Laptop GPU")
|
|
== "GeForce RTX 5090 Mobile"
|
|
)
|
|
assert _normalize_detected_name("Intel(R) Arc(TM) A770 Graphics") == "Arc A770"
|
|
assert _normalize_detected_name("AMD Radeon RX 6750 XT") == "Radeon RX 6750 XT"
|
|
|
|
|
|
def test_normalize_empty_or_vendor_only():
|
|
assert _normalize_detected_name("") == ""
|
|
assert _normalize_detected_name("AMD") == ""
|
|
|
|
|
|
def test_normalize_adds_space_to_vram_bin_suffix():
|
|
# #98: drivers write "12GB", dbgpu writes "12 GB".
|
|
assert _normalize_detected_name("NVIDIA RTX A2000 12GB") == "RTX A2000 12 GB"
|
|
|
|
|
|
# --- curated lookup is mobile-aware ---------------------------------------
|
|
|
|
|
|
def test_static_bandwidth_desktop_key_does_not_claim_laptop_card():
|
|
# "RTX 5090" desktop key (1792) must not match a Laptop driver name.
|
|
assert _static_bandwidth("NVIDIA GeForce RTX 5090 Laptop GPU") is None
|
|
# The desktop card itself still resolves from the curated table.
|
|
assert _static_bandwidth("NVIDIA GeForce RTX 5090") == 1792.0
|
|
|
|
|
|
def test_static_bandwidth_keeps_curated_laptop_entry():
|
|
# A curated key that is itself a laptop entry stays authoritative.
|
|
assert _static_bandwidth("NVIDIA RTX A3000 Laptop GPU") == 264.0
|
|
|
|
|
|
# --- end-to-end resolution -------------------------------------------------
|
|
|
|
|
|
def test_resolve_desktop_uses_curated_value():
|
|
assert resolve_detected_bandwidth("NVIDIA GeForce RTX 5090") == 1792.0
|
|
|
|
|
|
def test_resolve_laptop_5090_is_mobile_not_desktop():
|
|
# The bug in #74: a laptop 5090 must not inherit the 1792 desktop value.
|
|
bw = resolve_detected_bandwidth("NVIDIA GeForce RTX 5090 Laptop GPU", 24 * _GiB)
|
|
assert bw is not None
|
|
assert bw < 1500.0 # mobile 5090 is ~896 GB/s, nowhere near desktop 1792
|
|
|
|
|
|
def test_resolve_a3000_laptop_preserves_curated_264():
|
|
assert resolve_detected_bandwidth("NVIDIA RTX A3000 Laptop GPU", 6 * _GiB) == 264.0
|
|
|
|
|
|
def test_resolve_rx_6750_xt():
|
|
# #61: originally None. The curated table now carries it (#68); dbgpu
|
|
# would agree (432 GB/s) if the curated entry ever went away.
|
|
bw = resolve_detected_bandwidth("AMD Radeon RX 6750 XT", 12 * _GiB)
|
|
assert bw == 432.0
|
|
|
|
|
|
def test_resolve_variant_qualifier_is_preserved():
|
|
# "RTX 4060 Ti" must not collapse to the non-Ti "RTX 4060".
|
|
bw = resolve_detected_bandwidth("NVIDIA GeForce RTX 4060 Ti", 16 * _GiB)
|
|
assert bw is not None
|
|
# 4060 (non-Ti) is 272 GB/s; the Ti is 288. Either way it must be a real,
|
|
# positive value and not desktop-flagship sized.
|
|
assert 200 < bw < 400
|
|
|
|
|
|
def test_resolve_arc_pro_b70_uses_curated_value():
|
|
assert resolve_detected_bandwidth("Intel(R) Arc(TM) Pro B70 Graphics") == 608.0
|
|
assert resolve_detected_bandwidth("Battlemage G31 [Intel Graphics]") == 608.0
|
|
|
|
|
|
def test_resolve_empty_name_returns_none():
|
|
assert resolve_detected_bandwidth("") is None
|
|
assert resolve_detected_bandwidth("Some Totally Made Up GPU 9xZ") is None
|
|
|
|
|
|
def test_resolve_known_amd_desktop_from_curated_table():
|
|
assert resolve_detected_bandwidth("AMD Radeon RX 9070 XT") == 640.0
|
|
|
|
|
|
def test_resolve_a2000_12gb_vram_bin_from_dbgpu():
|
|
# #98: "NVIDIA RTX A2000 12GB" reported BW: N/A. dbgpu names the bin
|
|
# "RTX A2000 12 GB" (288 GB/s); the normalizer must bridge the spacing.
|
|
assert resolve_detected_bandwidth("NVIDIA RTX A2000 12GB", 12 * _GiB) == 288.0
|
|
|
|
|
|
def test_static_bandwidth_compound_lspci_name():
|
|
# Ported from amd.py (#68): compound lspci names resolve via segment
|
|
# splitting, first matching segment wins, "RX " prefix retried for bare
|
|
# segments.
|
|
compound = "Navi 22 [Radeon RX 6700/6700 XT/6750 XT / 6800M/6850M XT]"
|
|
bw = _static_bandwidth(compound)
|
|
assert bw is not None
|
|
assert bw > 0
|
|
# The same answer flows through the public resolver.
|
|
assert resolve_detected_bandwidth(compound, 12 * _GiB) == bw
|