diff options
Diffstat (limited to 'src/app/data_harvester/memory/gpu.rs')
-rw-r--r-- | src/app/data_harvester/memory/gpu.rs | 47 |
1 files changed, 0 insertions, 47 deletions
diff --git a/src/app/data_harvester/memory/gpu.rs b/src/app/data_harvester/memory/gpu.rs deleted file mode 100644 index 6fd66ba3..00000000 --- a/src/app/data_harvester/memory/gpu.rs +++ /dev/null @@ -1,47 +0,0 @@ -use super::MemHarvest; - -/// Return GPU memory usage. -#[cfg(feature = "gpu")] -pub(crate) fn get_gpu_mem_usage() -> Option<Vec<(String, MemHarvest)>> { - // As we add more support, expand on this. - - #[cfg(feature = "nvidia")] - get_nvidia_mem_usage() -} - -/// Returns the memory usage of NVIDIA cards. -#[inline] -#[cfg(feature = "nvidia")] -fn get_nvidia_mem_usage() -> Option<Vec<(String, MemHarvest)>> { - use crate::data_harvester::nvidia::NVML_DATA; - - if let Ok(nvml) = &*NVML_DATA { - if let Ok(num_gpu) = nvml.device_count() { - let mut results = Vec::with_capacity(num_gpu as usize); - for i in 0..num_gpu { - if let Ok(device) = nvml.device_by_index(i) { - if let (Ok(name), Ok(mem)) = (device.name(), device.memory_info()) { - // add device memory in bytes - results.push(( - name, - MemHarvest { - total_bytes: mem.total, - used_bytes: mem.used, - use_percent: if mem.total == 0 { - None - } else { - Some(mem.used as f64 / mem.total as f64 * 100.0) - }, - }, - )); - } - } - } - Some(results) - } else { - None - } - } else { - None - } -} |