Skip to content

Commit 43d7e5b

Browse files
committed
fix macos build issues by specifying enzyme lib dir
1 parent 4d3b2cd commit 43d7e5b

File tree

2 files changed

+5
-1
lines changed

2 files changed

+5
-1
lines changed

src/bootstrap/src/core/build_steps/compile.rs

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1174,6 +1174,10 @@ pub fn rustc_cargo(
11741174
// We want to link against registerEnzyme and in the future we want to use additional
11751175
// functionality from Enzyme core. For that we need to link against Enzyme.
11761176
if builder.config.llvm_enzyme {
1177+
let arch = builder.build.build;
1178+
let enzyme_dir = builder.build.out.join(arch).join("enzyme").join("lib");
1179+
cargo.rustflag("-L").rustflag(enzyme_dir.to_str().expect("Invalid path"));
1180+
11771181
let llvm_config = builder.llvm_config(builder.config.build).unwrap();
11781182
let llvm_version_major = llvm::get_llvm_version_major(builder, &llvm_config);
11791183
cargo.rustflag("-l").rustflag(&format!("Enzyme-{llvm_version_major}"));

tests/codegen/autodiff.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
//@ compile-flags: -C opt-level=3 -Clto=fat
1+
//@ compile-flags: -Zautodiff=Enable -C opt-level=3 -Clto=fat
22
//@ no-prefer-dynamic
33
//@ needs-enzyme
44
#![feature(autodiff)]

0 commit comments

Comments
 (0)