fix(sys): link xnnpack microkernels

why
pykeio · Oct 23, 2024 · 585efda · 585efda
1 parent 2628378
commit 585efda
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/ort-sys/build.rs b/ort-sys/build.rs
@@ -344,8 +344,11 @@ fn prepare_libort_dir() -> (PathBuf, bool) {
 					optional_link_lib(&lib_dir, "onnxruntime_providers_rknpu");
 					optional_link_lib(&lib_dir, "onnxruntime_providers_tvm");
 					if optional_link_lib(&lib_dir, "onnxruntime_providers_xnnpack") {
-						add_search_dir(transform_dep(external_lib_dir.join("googlexnnpack-build"), &profile));
+						let xnnpack_build_dir = transform_dep(external_lib_dir.join("googlexnnpack-build"), &profile);
+						add_search_dir(&xnnpack_build_dir);
 						println!("cargo:rustc-link-lib=static=XNNPACK");
+						optional_link_lib(&xnnpack_build_dir, "microkernels-prod");
+
 						add_search_dir(transform_dep(external_lib_dir.join("pthreadpool-build"), &profile));
 						println!("cargo:rustc-link-lib=static=pthreadpool");
 					}