Fix compile error on macos arm64.

frozenlib · Apr 28, 2024 · 7510e86 · 7510e86
1 parent 0ee76bf
commit 7510e86
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
@@ -65,7 +65,7 @@ jobs:
       - name: Rustup update stable
         run: rustup update stable
       - name: Show cargo version
-        run: cargo --version
+        run: cargo --version -v
       - name: Build
         run: cargo build --verbose
       - name: Build tests

diff --git a/lgbm-sys/build.rs b/lgbm-sys/build.rs
@@ -50,7 +50,12 @@ fn build_linux() {
     }
 }
 fn build_macos() {
-    println!("cargo:rustc-link-lib=dylib=_lightgbm");
+    let target = env::var("CARGO_BUILD_TARGET").unwrap();
+    if target.contains("aarch64") {
+        println!("cargo:rustc-link-lib=dylib=lightgbm");
+    } else {
+        println!("cargo:rustc-link-lib=dylib=_lightgbm");
+    }
 }
 
 fn env_var(key: &str) -> String {