add enzyme distribution step

Author: Shourya742

src/bootstrap/src/core/build_steps/dist.rs

@@ -2599,3 +2599,38 @@ impl Step for Gcc {
         tarball.generate()
     }
 }
+
+#[derive(Clone, Debug, Eq, Hash, PartialEq)]
+pub struct Enzyme {
+    pub target: TargetSelection,
+}
+
+impl Step for Enzyme {
+    type Output = GeneratedTarball;
+    const DEFAULT: bool = false;
+    const ONLY_HOSTS: bool = true;
+
+    fn should_run(run: ShouldRun<'_>) -> ShouldRun<'_> {
+        run.alias("enzyme")
+    }
+
+    fn make_run(run: RunConfig<'_>) {
+        run.builder.ensure(Enzyme { target: run.target });
+    }
+
+    fn run(self, builder: &Builder<'_>) -> Self::Output {
+        let mut tarball = Tarball::new(builder, "enzyme", &self.target.triple);
+        let enzyme = builder.ensure(super::llvm::Enzyme { target: self.target });
+        tarball.set_overlay(OverlayKind::Enzyme);
+        tarball.is_preview(true);
+        if let Some(llvm_config) = builder.llvm_config(builder.config.build) {
+            let llvm_version_major = llvm::get_llvm_version_major(builder, &llvm_config);
+            let lib_ext = std::env::consts::DLL_EXTENSION;
+            let libenzyme = format!("libEnzyme-{llvm_version_major}");
+            let src_lib = enzyme.join("build/Enzyme").join(&libenzyme).with_extension(lib_ext);
+            tarball.add_file(src_lib, "", FileType::NativeLibrary);
+        }
+        tarball.add_legal_and_readme_to("share/doc/enzyme");
+        tarball.generate()
+    }
+}

src/bootstrap/src/core/builder/mod.rs

@@ -1084,7 +1084,8 @@ impl<'a> Builder<'a> {
                 dist::PlainSourceTarball,
                 dist::BuildManifest,
                 dist::ReproducibleArtifacts,
-                dist::Gcc
+                dist::Gcc,
+                dist::Enzyme
             ),
             Kind::Install => describe!(
                 install::Docs,

src/bootstrap/src/utils/tarball.rs

@@ -26,6 +26,7 @@ pub(crate) enum OverlayKind {
     RustAnalyzer,
     RustcCodegenCranelift,
     LlvmBitcodeLinker,
+    Enzyme
 }
 
 impl OverlayKind {
@@ -72,6 +73,7 @@ impl OverlayKind {
                 "LICENSE-MIT",
                 "src/tools/llvm-bitcode-linker/README.md",
             ],
+            OverlayKind::Enzyme => &["src/tools/enzyme/Readme.md", "src/tools/enzyme/LICENSE"]
         }
     }
 
@@ -94,6 +96,7 @@ impl OverlayKind {
                 .version(builder, &builder.release_num("rust-analyzer/crates/rust-analyzer")),
             OverlayKind::RustcCodegenCranelift => builder.rust_version(),
             OverlayKind::LlvmBitcodeLinker => builder.rust_version(),
+            OverlayKind::Enzyme => builder.rust_version()
         }
     }
 }