Elazarl/feature/cuuuid

src/driver/result.rs

@@ -173,6 +173,16 @@ pub mod device {
         let name = CStr::from_bytes_until_nul(&buf).expect("No null byte was present");
         Ok(String::from_utf8_lossy(name.to_bytes()).into())
     }
+
+    pub fn get_uuid(dev: sys::CUdevice) -> Result<sys::CUuuid, DriverError> {
+        let id: sys::CUuuid;
+        unsafe {
+            let mut uuid = MaybeUninit::uninit();
+            lib().cuDeviceGetUuid(uuid.as_mut_ptr(), dev).result()?;
+            id = uuid.assume_init();
+        }
+        Ok(id)
+    }
 }
 
 pub mod function {

src/driver/safe/core.rs

@@ -134,6 +134,11 @@ impl CudaDevice {
         result::device::get_name(self.cu_device)
     }
 
+    /// Get the UUID of this device.
+    pub fn uuid(&self) -> Result<sys::CUuuid, result::DriverError> {
+        result::device::get_uuid(self.cu_device)
+    }
+
     /// Get the underlying [sys::CUdevice] of this [CudaDevice].
     ///
     /// # Safety

src/lib.rs

@@ -87,12 +87,12 @@ pub mod cudnn;
 pub mod curand;
 #[cfg(feature = "driver")]
 pub mod driver;
-#[cfg(feature = "runtime")]
-pub mod runtime;
 #[cfg(feature = "nccl")]
 pub mod nccl;
 #[cfg(feature = "nvrtc")]
 pub mod nvrtc;
+#[cfg(feature = "runtime")]
+pub mod runtime;
 
 pub mod types;