Do not use the as-needed flag on macos.

LaurentMazare · Mar 11, 2023 · ce48283 · ce48283
1 parent c90ee79
commit ce48283
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/build.rs b/build.rs
@@ -1,5 +1,11 @@
 fn main() {
-    println!("cargo:rustc-link-arg=-Wl,--no-as-needed");
-    println!("cargo:rustc-link-arg=-Wl,--copy-dt-needed-entries");
-    println!("cargo:rustc-link-arg=-ltorch");
+    let os = std::env::var("CARGO_CFG_TARGET_OS").expect("Unable to get TARGET_OS");
+    match os.as_str() {
+        "linux" | "windows" => {
+            println!("cargo:rustc-link-arg=-Wl,--no-as-needed");
+            println!("cargo:rustc-link-arg=-Wl,--copy-dt-needed-entries");
+            println!("cargo:rustc-link-arg=-ltorch");
+        }
+        _ => {}
+    }
 }