fix macos builds (#168)

Author: ZuseZ4

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

@@ -1776,25 +1776,25 @@ impl Step for Assemble {
         //        use that to bootstrap this compiler forward.
         let mut build_compiler = builder.compiler(target_compiler.stage - 1, builder.config.build);
 
-
         // Build enzyme
-        let enzyme_install =
-            Some(builder.ensure(llvm::Enzyme { target: build_compiler.host }));
-        //let enzyme_install = if builder.config.llvm_enzyme {
-        //    Some(builder.ensure(llvm::Enzyme { target: build_compiler.host }))
-        //} else {
-        //    None
-        //};
+        let enzyme_install = if builder.config.llvm_enzyme {
+            Some(builder.ensure(llvm::Enzyme { target: build_compiler.host }))
+        } else {
+            None
+        };
 
         if let Some(enzyme_install) = enzyme_install {
-            let src_lib = enzyme_install.join("build/Enzyme/LLVMEnzyme-19.so");
+            let lib_ext = match env::consts::OS {
+                "macos" => "dylib",
+                "windows" => "dll",
+                _ => "so",
+            };
 
+            let src_lib = enzyme_install.join("build/Enzyme/libEnzyme-19").with_extension(lib_ext);
             let libdir = builder.sysroot_libdir(build_compiler, build_compiler.host);
             let target_libdir = builder.sysroot_libdir(target_compiler, target_compiler.host);
-            let dst_lib = libdir.join("libLLVMEnzyme-19.so");
-            let target_dst_lib = target_libdir.join("libLLVMEnzyme-19.so");
-            //builder.copy_extra_objects(builder, &compiler, target);
-            //builder.copy_extra_objects(builder, &compiler, target);
+            let dst_lib = libdir.join("libEnzyme-19").with_extension(lib_ext);
+            let target_dst_lib = target_libdir.join("libEnzyme-19").with_extension(lib_ext);
             builder.copy_link(&src_lib, &dst_lib);
             builder.copy_link(&src_lib, &target_dst_lib);
         }

src/bootstrap/src/core/build_steps/llvm.rs

@@ -911,7 +911,7 @@ impl Step for Enzyme {
             .profile(profile)
             .env("LLVM_CONFIG_REAL", &llvm_config)
             .define("LLVM_ENABLE_ASSERTIONS", "ON")
-            .define("ENZYME_EXTERNAL_SHARED_LIB", "OFF")
+            .define("ENZYME_EXTERNAL_SHARED_LIB", "ON")
             .define("LLVM_DIR", builder.llvm_out(target));
 
         cfg.build();

src/bootstrap/src/core/builder.rs

@@ -1585,7 +1585,7 @@ impl<'a> Builder<'a> {
 
         // https://rust-lang.zulipchat.com/#narrow/stream/182449-t-compiler.2Fhelp/topic/.E2.9C.94.20link.20new.20library.20into.20stage1.2Frustc
         rustflags.arg("-l");
-        rustflags.arg("LLVMEnzyme-19");
+        rustflags.arg("Enzyme-19");
 
         let use_new_symbol_mangling = match self.config.rust_new_symbol_mangling {
             Some(setting) => {