#217 Replace unwrap with ? in CudaDevice::new() (#223)

coreylowman · May 1, 2024 · d565d8c · d565d8c
1 parent 64ddbc7
commit d565d8c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/driver/safe/core.rs b/src/driver/safe/core.rs
@@ -51,9 +51,9 @@ unsafe impl Sync for CudaDevice {}
 impl CudaDevice {
     /// Creates a new [CudaDevice] on device index `ordinal`.
     pub fn new(ordinal: usize) -> Result<Arc<Self>, result::DriverError> {
-        result::init().unwrap();
+        result::init()?;
 
-        let cu_device = result::device::get(ordinal as i32).unwrap();
+        let cu_device = result::device::get(ordinal as i32)?;
 
         // primary context initialization, can fail with OOM
         let cu_primary_ctx = unsafe { result::primary_ctx::retain(cu_device) }?;