wehub-resource-sync
57 lines
1.7 KiB
Python
57 lines
1.7 KiB
Python
"""Unified hardware detection orchestrator."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import logging
|
|
import platform
|
|
|
|
from whichllm.hardware.amd import detect_amd_gpus
|
|
from whichllm.hardware.apple import detect_apple_gpu, detect_apple_gpu_linux
|
|
from whichllm.hardware.cpu import detect_avx_support, detect_cpu_cores, detect_cpu_name
|
|
from whichllm.hardware.intel import detect_intel_gpus
|
|
from whichllm.hardware.memory import detect_disk_free_bytes, detect_ram_bytes
|
|
from whichllm.hardware.nvidia import detect_nvidia_gpus
|
|
from whichllm.hardware.types import HardwareInfo
|
|
from whichllm.hardware.windows import detect_windows_gpus
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
def detect_hardware() -> HardwareInfo:
|
|
"""Detect all hardware. Each detector is fail-safe (returns empty on error)."""
|
|
os_name = platform.system().lower()
|
|
if os_name not in ("linux", "darwin", "windows"):
|
|
os_name = "linux"
|
|
|
|
# GPU detection
|
|
gpus = []
|
|
gpus.extend(detect_nvidia_gpus())
|
|
if os_name == "linux":
|
|
gpus.extend(detect_amd_gpus())
|
|
gpus.extend(detect_intel_gpus())
|
|
gpus.extend(detect_apple_gpu_linux())
|
|
if os_name == "darwin":
|
|
gpus.extend(detect_apple_gpu())
|
|
if os_name == "windows":
|
|
gpus.extend(detect_windows_gpus())
|
|
|
|
# CPU
|
|
cpu_name = detect_cpu_name()
|
|
cpu_cores = detect_cpu_cores()
|
|
has_avx2, has_avx512 = detect_avx_support()
|
|
|
|
# Memory
|
|
ram_bytes = detect_ram_bytes()
|
|
disk_free = detect_disk_free_bytes()
|
|
|
|
return HardwareInfo(
|
|
gpus=gpus,
|
|
cpu_name=cpu_name,
|
|
cpu_cores=cpu_cores,
|
|
has_avx2=has_avx2,
|
|
has_avx512=has_avx512,
|
|
ram_bytes=ram_bytes,
|
|
disk_free_bytes=disk_free,
|
|
os=os_name,
|
|
)
|