Disable CUDA provider options.

robohouse-delft · Nov 23, 2023 · 094b04e · 094b04e
1 parent 16c7d90
commit 094b04e
Showing 1 changed file with 15 additions and 15 deletions.
diff --git a/rust/onnxruntime/src/session.rs b/rust/onnxruntime/src/session.rs
@@ -164,14 +164,14 @@ impl<'a> SessionBuilder<'a> {
     /// Append a CUDA execution provider
     pub fn with_execution_provider_cuda(self) -> Result<SessionBuilder<'a>> {
         let mut cuda_options: *mut sys::OrtCUDAProviderOptionsV2 = null_mut();
-        let status = unsafe {
-            self.env
-                .env()
-                .api()
-                .CreateCUDAProviderOptions
-                .unwrap()(&mut cuda_options)
-        };
-        status_to_result(status).map_err(OrtError::CudaProviderOptions)?;
+        // let status = unsafe {
+        //     self.env
+        //         .env()
+        //         .api()
+        //         .CreateCUDAProviderOptions
+        //         .unwrap()(&mut cuda_options)
+        // };
+        // status_to_result(status).map_err(OrtError::CudaProviderOptions)?;
 
         let status = unsafe {
             self.env
@@ -182,13 +182,13 @@ impl<'a> SessionBuilder<'a> {
         };
         status_to_result(status).map_err(OrtError::AppendExecutionProviderCuda)?;
 
-        unsafe {
-            self.env
-                .env()
-                .api()
-                .ReleaseCUDAProviderOptions
-                .unwrap()(cuda_options);
-        };
+        // unsafe {
+        //     self.env
+        //         .env()
+        //         .api()
+        //         .ReleaseCUDAProviderOptions
+        //         .unwrap()(cuda_options);
+        // };
         Ok(self)
     }