Skip to main content

neuronbox_runtime/host/
mod.rs

1//! Single host probe layer: GPU, platform, likely PyTorch backend.
2
3mod apple;
4mod nvidia;
5#[cfg(all(target_os = "linux", feature = "nvml"))]
6mod nvml_linux;
7mod probe;
8mod rocm;
9mod snapshot;
10
11pub use nvidia::{compute_apps_display_lines, compute_apps_pid_memory_mb};
12pub use probe::HostProbe;
13pub use snapshot::{
14    GpuRecord, HostSnapshot, PlatformInfo, ProbeStatus, TrainingBackend,
15    HOST_SNAPSHOT_SCHEMA_VERSION,
16};