fix: bundle libonnxruntime in rlib/staticlib builds (#215)

Fixes #214.
pykeio · Jun 21, 2024 · 882f657 · 882f657
1 parent c64b8ea
commit 882f657
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/ort-sys/build.rs b/ort-sys/build.rs
@@ -409,7 +409,15 @@ fn real_main(link: bool) {
 
 	if link {
 		if needs_link {
-			println!("cargo:rustc-link-lib=onnxruntime");
+			let target_os = env::var("CARGO_CFG_TARGET_OS").unwrap();
+			let static_lib_file_name = if target_os.contains("windows") { "onnxruntime.lib" } else { "libonnxruntime.a" };
+
+			let static_lib_path = lib_dir.join(static_lib_file_name);
+			if static_lib_path.exists() {
+				println!("cargo:rustc-link-lib=static=onnxruntime");
+			} else {
+				println!("cargo:rustc-link-lib=onnxruntime");
+			}
 			println!("cargo:rustc-link-search=native={}", lib_dir.display());
 		}