add debug output

fluencelabs · Sep 9, 2024 · 7fe3cc2 · 7fe3cc2
1 parent aaa08cf
commit 7fe3cc2
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/crates/gpu-utils/src/lib.rs b/crates/gpu-utils/src/lib.rs
@@ -38,6 +38,8 @@ pub fn get_gpu_pci() -> Result<HashSet<PciLocation>, PciError> {
         pci_devices.insert(device_location, device_class);
     }
 
+    tracing::debug!(target: "gpu-utils", "Found GPU devices: {:?}", gpu_devices);
+
     let result = match get_iommu_groups() {
         Ok(iommu_groups) => {
             // Find all devices that are in the same IOMMU group as the GPU devices
@@ -57,6 +59,9 @@ pub fn get_gpu_pci() -> Result<HashSet<PciLocation>, PciError> {
             gpu_devices
         }
     };
+
+    tracing::debug!(target: "gpu-utils", "Importing PCI devices: {:?}", result);
+
     Ok(result)
 }