refactor(transformer-cpu): 开启 mkl 支持，自动检测

Signed-off-by: YdrMaster <[email protected]>
InfiniTensor · Mar 21, 2024 · bcc8b68 · bcc8b68
1 parent 6286fea
commit bcc8b68
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 8 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/transformer-cpu/Cargo.toml b/transformer-cpu/Cargo.toml
@@ -12,3 +12,6 @@ tensor = { path = "../tensor" }
 transformer = { path = "../transformer" }
 gemm = "0.17"
 intel-mkl-src = "0.8"
+
+[build-dependencies]
+intel-mkl-tool = "0.8"
diff --git a/transformer-cpu/build.rs b/transformer-cpu/build.rs
@@ -1,4 +1,7 @@
 fn main() {
-    // println!("cargo:rustc-cfg=detected_mkl");
-    // println!("cargo:rustc-link-lib=dylib=mkl_rt");
+    if !intel_mkl_tool::Library::available().is_empty() {
+        // **NOTICE** add mkl_rt to library search path
+        println!("cargo:rustc-cfg=detected_mkl");
+        println!("cargo:rustc-link-lib=dylib=mkl_rt");
+    }
 }