diff options
author | Clement Tsang <34804052+ClementTsang@users.noreply.github.com> | 2023-02-27 19:37:35 -0500 |
---|---|---|
committer | GitHub <noreply@github.com> | 2023-02-27 19:37:35 -0500 |
commit | 44e52d622991ebbfe3939b126a3aa316e0f21723 (patch) | |
tree | 54dc2d10dde867e03dc7606a61a389ebae5ddf52 /src | |
parent | 6fa72290a952f3f34867769f3d3f6c72b3655e6b (diff) |
deps: bump nvml to 0.9.0 (#1032)
Diffstat (limited to 'src')
-rw-r--r-- | src/app/data_harvester/memory/general/heim.rs | 2 | ||||
-rw-r--r-- | src/app/data_harvester/nvidia.rs | 1 |
2 files changed, 3 insertions, 0 deletions
diff --git a/src/app/data_harvester/memory/general/heim.rs b/src/app/data_harvester/memory/general/heim.rs index bb63d2b5..27221420 100644 --- a/src/app/data_harvester/memory/general/heim.rs +++ b/src/app/data_harvester/memory/general/heim.rs @@ -257,9 +257,11 @@ pub async fn get_arc_data() -> crate::utils::error::Result<Option<MemHarvest>> { })) } +// FIXME: Can unify this with the sysinfo impl #[cfg(feature = "nvidia")] pub async fn get_gpu_data() -> crate::utils::error::Result<Option<Vec<(String, MemHarvest)>>> { use crate::data_harvester::nvidia::NVML_DATA; + if let Ok(nvml) = &*NVML_DATA { if let Ok(ngpu) = nvml.device_count() { let mut results = Vec::with_capacity(ngpu as usize); diff --git a/src/app/data_harvester/nvidia.rs b/src/app/data_harvester/nvidia.rs index 2cb4b622..9619da64 100644 --- a/src/app/data_harvester/nvidia.rs +++ b/src/app/data_harvester/nvidia.rs @@ -1,3 +1,4 @@ use nvml_wrapper::{error::NvmlError, Nvml}; use once_cell::sync::Lazy; + pub static NVML_DATA: Lazy<Result<Nvml, NvmlError>> = Lazy::new(Nvml::init); |