bumped version to 0.5.2: bind through Device::set only (no guard)

Cargo.toml

@@ -5,13 +5,13 @@ keywords = ["async", "nvidia", "cuda", "gpu", "tensorrt"]
 readme = "README.md"
 categories = ["asynchronous"]
 edition = "2021"
-version = "0.5.1"
+version = "0.5.2"
 authors = ["Oddity.ai Developers <hello@oddity.ai>"]
 repository = "https://github.com/oddity-ai/async-tensorrt"
 license = "MIT OR Apache-2.0"
 
 [dependencies]
-async-cuda = "0.5.3"
+async-cuda = "0.5.4"
 cpp = "0.5"
 tracing = "0.1"
 

src/ffi/sync/builder.rs

@@ -34,8 +34,7 @@ unsafe impl Sync for Builder {}
 
 impl Builder {
     pub fn new() -> Self {
-        let device =
-            Device::get().unwrap_or_else(|err| panic!("failed to get current device: {err}"));
+        let device = Device::get_or_panic();
         let addr = cpp!(unsafe [] -> *mut std::ffi::c_void as "void*" {
             return createInferBuilder(GLOBAL_LOGGER);
         });
@@ -141,7 +140,7 @@ impl Builder {
 
 impl Drop for Builder {
     fn drop(&mut self) {
-        let _device_guard = Device::bind_or_panic(self.device);
+        Device::set_or_panic(self.device);
         let internal = self.as_mut_ptr();
         cpp!(unsafe [
             internal as "void*"

src/ffi/sync/engine.rs

@@ -130,7 +130,7 @@ impl Engine {
 
 impl Drop for Engine {
     fn drop(&mut self) {
-        let _device_guard = Device::bind_or_panic(self.runtime.device());
+        Device::set_or_panic(self.runtime.device());
         let Engine { internal, .. } = *self;
         cpp!(unsafe [
             internal as "void*"
@@ -262,7 +262,7 @@ impl<'engine> ExecutionContext<'engine> {
     }
 
     unsafe fn new_internal(engine: &mut Engine) -> *mut std::ffi::c_void {
-        let _device_guard = Device::bind_or_panic(engine.device());
+        Device::set_or_panic(engine.device());
         let internal_engine = engine.as_mut_ptr();
         let internal = cpp!(unsafe [
             internal_engine as "void*"
@@ -301,7 +301,7 @@ impl<'engine> ExecutionContext<'engine> {
 
 impl<'engine> Drop for ExecutionContext<'engine> {
     fn drop(&mut self) {
-        let _device_guard = Device::bind_or_panic(self.device);
+        Device::set_or_panic(self.device);
         let ExecutionContext { internal, .. } = *self;
         cpp!(unsafe [
             internal as "void*"

src/ffi/sync/runtime.rs

@@ -33,8 +33,7 @@ unsafe impl Sync for Runtime {}
 
 impl Runtime {
     pub fn new() -> Self {
-        let device =
-            Device::get().unwrap_or_else(|err| panic!("failed to get current device: {err}"));
+        let device = Device::get_or_panic();
         let addr = cpp!(unsafe [] -> *mut std::ffi::c_void as "void*" {
             return createInferRuntime(GLOBAL_LOGGER);
         });
@@ -74,7 +73,7 @@ impl Runtime {
         buffer_ptr: *const std::ffi::c_void,
         buffer_size: usize,
     ) -> Result<Engine> {
-        Device::bind(self.device)?;
+        Device::set(self.device)?;
         let internal = self.as_mut_ptr();
         let internal_engine = cpp!(unsafe [
             internal as "void*",
@@ -104,7 +103,7 @@ impl Runtime {
 
 impl Drop for Runtime {
     fn drop(&mut self) {
-        let _device_guard = Device::bind_or_panic(self.device);
+        Device::set_or_panic(self.device);
         let internal = self.as_mut_ptr();
         cpp!(unsafe [
             internal as "void*"