Do not error if EP is not avaliable, include more eps in cargo.toml

Author: [email protected]

native/ortex/Cargo.toml

@@ -15,7 +15,13 @@ resolver = "2"
 [dependencies]
 rustler = "0.36.2"
 ort-sys = { version = "=2.0.0-rc.10", default-features = false }
-ort = { version = "2.0.0-rc.10", features = ["half", "cuda"] }
+ort = { version = "2.0.0-rc.10", features = [
+    "half",
+    "cuda",
+    "tensorrt",
+    "directml",
+    "coreml"
+] }
 ndarray = "0.16.1"
 half = "2.6.0"
 tracing-subscriber = { version = "0.3", features = ["env-filter", "fmt"] }

native/ortex/src/utils.rs

@@ -7,8 +7,8 @@ use ndarray::{ArrayViewMut, Ix, IxDyn};
 
 use ndarray::ShapeError;
 
-use rustler::ResourceArc;
 use rustler::types::Binary;
+use rustler::ResourceArc;
 use rustler::{Atom, Env, NifResult};
 
 use ort::execution_providers::ExecutionProviderDispatch;
@@ -96,7 +96,7 @@ pub fn map_eps(env: rustler::env::Env, eps: Vec<Atom>) -> Vec<ExecutionProviderD
     eps.iter()
         .map(|e| match &e.to_term(env).atom_to_string().unwrap()[..] {
             CPU => ort::execution_providers::cpu::CPUExecutionProvider::default().build(),
-            CUDA => ort::execution_providers::cuda::CUDAExecutionProvider::default().build().error_on_failure(),
+            CUDA => ort::execution_providers::cuda::CUDAExecutionProvider::default().build(),
             TENSORRT => {
                 ort::execution_providers::tensorrt::TensorRTExecutionProvider::default().build()
             }