From 8988d15745d3f7a3591797b7f69996ed4ee33be5 Mon Sep 17 00:00:00 2001 From: "Carson M." Date: Sun, 20 Oct 2024 11:51:44 -0500 Subject: [PATCH] fix(sys): link to static EPs, fixes #306 --- ort-sys/build.rs | 13 +++++++++++-- 1 file changed, 11 insertions(+), 2 deletions(-) diff --git a/ort-sys/build.rs b/ort-sys/build.rs index d2f21fe0..40017eac 100644 --- a/ort-sys/build.rs +++ b/ort-sys/build.rs @@ -328,8 +328,17 @@ fn prepare_libort_dir() -> (PathBuf, bool) { println!("cargo:rustc-link-lib=onnxruntime_providers_coreml"); } - // #[cfg(feature = "rocm")] - // println!("cargo:rustc-link-lib=onnxruntime_providers_rocm"); + // link static EPs if present + // not sure if these are the right libs but they're optional links so... + optional_link_lib(&lib_dir, "onnxruntime_providers_acl"); + optional_link_lib(&lib_dir, "onnxruntime_providers_armnn"); + optional_link_lib(&lib_dir, "onnxruntime_providers_azure"); + optional_link_lib(&lib_dir, "onnxruntime_providers_dml"); + optional_link_lib(&lib_dir, "onnxruntime_providers_nnapi"); + optional_link_lib(&lib_dir, "onnxruntime_providers_qnn"); + optional_link_lib(&lib_dir, "onnxruntime_providers_rknpu"); + optional_link_lib(&lib_dir, "onnxruntime_providers_tvm"); + optional_link_lib(&lib_dir, "onnxruntime_providers_xnnpack"); needs_link = false; break;