Merge pull request #207 from kusaanko/fix-openblas

Fix to be able to compile with openblas
This commit is contained in:
Niko 2025-03-02 10:43:25 -07:00 committed by GitHub
commit 95c4423ec9
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -214,6 +214,12 @@ fn main() {
if cfg!(feature = "openblas") { if cfg!(feature = "openblas") {
config.define("GGML_BLAS", "ON"); config.define("GGML_BLAS", "ON");
config.define("GGML_BLAS_VENDOR", "OpenBLAS");
if env::var("BLAS_INCLUDE_DIRS").is_err() {
panic!("BLAS_INCLUDE_DIRS environment variable must be set when using OpenBLAS");
}
config.define("BLAS_INCLUDE_DIRS", env::var("BLAS_INCLUDE_DIRS").unwrap());
println!("cargo:rerun-if-env-changed=BLAS_INCLUDE_DIRS");
} }
if cfg!(feature = "metal") { if cfg!(feature = "metal") {
@ -255,7 +261,7 @@ fn main() {
println!("cargo:rustc-link-lib=static=ggml"); println!("cargo:rustc-link-lib=static=ggml");
println!("cargo:rustc-link-lib=static=ggml-base"); println!("cargo:rustc-link-lib=static=ggml-base");
println!("cargo:rustc-link-lib=static=ggml-cpu"); println!("cargo:rustc-link-lib=static=ggml-cpu");
if cfg!(target_os = "macos") { if cfg!(target_os = "macos") || cfg!(feature = "openblas") {
println!("cargo:rustc-link-lib=static=ggml-blas"); println!("cargo:rustc-link-lib=static=ggml-blas");
} }
if cfg!(feature = "vulkan") { if cfg!(feature = "vulkan") {