33 lines
1.2 KiB
Python
33 lines
1.2 KiB
Python
"""Vendor SoC recognizers derived from host facts.
|
|
|
|
Recognition uses OS-reported CPU or PCI identity without environment-variable input.
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
from hermes_platform.host import facts
|
|
|
|
# Match the CPU string because the chassis vendor does not identify the SoC.
|
|
_NVIDIA_SOC_VENDOR = "NVIDIA"
|
|
_NVIDIA_SOC_MODEL_MARKERS = ("N1X", "SPARK")
|
|
|
|
|
|
def is_nvidia_n1x_pci_id(pci_id: int) -> bool:
|
|
"""Recognize N1X; NVML packs device above the 16-bit vendor ID."""
|
|
return pci_id & 0xFFFF == 0x10DE and 0x2E00 <= pci_id >> 16 <= 0x2E3F
|
|
|
|
|
|
def looks_like_nvidia_arm_soc(*, native_arch: str, cpu_model: str, cpu_vendor: str) -> bool:
|
|
"""Return whether the supplied facts identify an NVIDIA ARM SoC."""
|
|
if native_arch != "arm64":
|
|
return False
|
|
model = cpu_model.upper()
|
|
vendor_match = _NVIDIA_SOC_VENDOR in model or cpu_vendor.strip().upper() == _NVIDIA_SOC_VENDOR
|
|
return vendor_match and any(marker in model for marker in _NVIDIA_SOC_MODEL_MARKERS)
|
|
|
|
|
|
def is_nvidia_arm_soc() -> bool:
|
|
"""Return whether this host has an NVIDIA ARM SoC."""
|
|
return looks_like_nvidia_arm_soc(
|
|
native_arch=facts.native_arch(), cpu_model=facts.cpu_model(), cpu_vendor=facts.cpu_vendor())
|