Add nvmlDeviceGetP2PStatus

nvml-wrapper/src/device.rs

@@ -3452,6 +3452,35 @@ impl<'nvml> Device<'nvml> {
         }
     }
 
+    /**
+    Gets the status for a given p2p capability index between this [`Device`] and another given [`Device`].
+
+    # Errors
+
+    * `Uninitialized`, if the library has not been successfully initialized
+    * `InvalidArg`, if device1 or device2 or p2p_index is invalid
+    * `Unknown`, on any unexpected error
+    */
+    #[doc(alias = "nvmlDeviceGetP2PStatus")]
+    pub fn p2p_status(
+        &self,
+        device2: &Device,
+        p2p_index: P2pCapabilitiesIndex,
+    ) -> Result<P2pStatus, NvmlError> {
+        let sym = nvml_sym(self.nvml.lib.nvmlDeviceGetP2PStatus.as_ref())?;
+
+        let status_c = unsafe {
+            let mut status: nvmlGpuP2PStatus_t = mem::zeroed();
+            let device2 = device2.device;
+
+            nvml_try(sym(self.device, device2, p2p_index as u32, &mut status))?;
+
+            status
+        };
+
+        P2pStatus::try_from(status_c)
+    }
+
     /**
     Gets the power source of this [`Device`].
 

nvml-wrapper/unwrapped_functions.txt

@@ -71,7 +71,6 @@ nvmlDeviceGetNvLinkRemotePciInfo_v2
 nvmlDeviceGetNvLinkUtilizationControl
 nvmlDeviceGetNvLinkUtilizationCounter
 nvmlDeviceGetOfaUtilization
-nvmlDeviceGetP2PStatus
 nvmlDeviceGetPgpuMetadataString
 nvmlDeviceGetPowerManagementDefaultLimit
 nvmlDeviceGetPowerManagementLimitConstraints