mirror of
https://github.com/pykeio/ort
synced 2026-04-25 16:34:55 +02:00
fix: web, no-std
This commit is contained in:
@@ -121,7 +121,7 @@ impl SessionBuilder {
|
||||
|
||||
let mut session_ptr = ptr::null_mut();
|
||||
let status = ortsys![unsafe CreateSession(self.environment.ptr(), model_url.as_ref(), self.ptr(), &mut session_ptr)].await;
|
||||
unsafe { crate::error::status_to_result(status) }?;
|
||||
unsafe { Error::result_from_status(status) }?;
|
||||
|
||||
let Some(session_ptr) = NonNull::new(session_ptr) else {
|
||||
return Err(Error::new(alloc::format!("Session creation failed with unknown error")));
|
||||
@@ -137,7 +137,7 @@ impl SessionBuilder {
|
||||
|
||||
let mut session_ptr = ptr::null_mut();
|
||||
let status = ortsys![unsafe CreateSessionFromArray(self.environment.ptr(), model_bytes.as_ref(), self.ptr(), &mut session_ptr)].await;
|
||||
unsafe { crate::error::status_to_result(status) }?;
|
||||
unsafe { Error::result_from_status(status) }?;
|
||||
|
||||
let Some(session_ptr) = NonNull::new(session_ptr) else {
|
||||
return Err(Error::new(alloc::format!("Session creation failed with unknown error")));
|
||||
|
||||
@@ -1,3 +1,5 @@
|
||||
use alloc::string::ToString;
|
||||
|
||||
use super::{BuilderResult, SessionBuilder};
|
||||
|
||||
// https://github.com/microsoft/onnxruntime/blob/main/include/onnxruntime/core/session/onnxruntime_session_options_config_keys.h
|
||||
|
||||
@@ -120,7 +120,7 @@ impl SessionBuilder {
|
||||
/// # }
|
||||
/// ```
|
||||
pub fn new() -> Result<Self> {
|
||||
let _environment = environment::current()?;
|
||||
let environment = environment::current()?;
|
||||
|
||||
let mut session_options_ptr: *mut ort_sys::OrtSessionOptions = ptr::null_mut();
|
||||
ortsys![unsafe CreateSessionOptions(&mut session_options_ptr)?; nonNull(session_options_ptr)];
|
||||
@@ -141,7 +141,7 @@ impl SessionBuilder {
|
||||
logger: None,
|
||||
no_global_thread_pool: false,
|
||||
no_env_eps: false,
|
||||
environment: _environment
|
||||
environment
|
||||
})
|
||||
}
|
||||
|
||||
|
||||
@@ -575,7 +575,7 @@ impl Session {
|
||||
)
|
||||
]
|
||||
.await;
|
||||
unsafe { crate::error::status_to_result(status) }?;
|
||||
unsafe { Error::result_from_status(status) }?;
|
||||
|
||||
let outputs = output_tensors
|
||||
.into_iter()
|
||||
|
||||
Reference in New Issue
Block a user