summaryrefslogtreecommitdiffstats
path: root/src/app/data_harvester/memory/gpu.rs
diff options
context:
space:
mode:
Diffstat (limited to 'src/app/data_harvester/memory/gpu.rs')
-rw-r--r--src/app/data_harvester/memory/gpu.rs47
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
- }
-}