The probe layer in src/gpu_gate/probe.py is already an interface (Probe) with an NVML implementation behind it. Adding an AmdSmiProbe and auto-detecting the vendor at startup would let the same wait/claim/run flow work on ROCm hosts without touching the selection logic.
The probe layer in
src/gpu_gate/probe.pyis already an interface (Probe) with an NVML implementation behind it. Adding anAmdSmiProbeand auto-detecting the vendor at startup would let the same wait/claim/run flow work on ROCm hosts without touching the selection logic.