48ed4e5364
Host facts and runtime predicates need one stdlib-only source that stays correct under architecture emulation. Hardware recognition reads host facts, not environment variables or GPU proxies. The package boundary gives later resolvers and catalog checks a shared foundation. Linear NS-920.
28 lines
1.0 KiB
Python
28 lines
1.0 KiB
Python
"""Vendor SoC recognizers derived from host facts.
|
|
|
|
Recognition uses OS-reported CPU 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 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())
|