chore/ci: Print if stamp is missing

.github/workflows/enzyme-ci.yml

@@ -0,0 +1,103 @@
+name: Rust CI
+
+on:
+  push:
+    branches:
+    - master
+  pull_request:
+    branches:
+    - master
+  merge_group:
+
+jobs:
+  build:
+    name: Rust Integration CI LLVM ${{ matrix.llvm }} ${{ matrix.build }} ${{ matrix.os }}
+    runs-on: ${{ matrix.os }}
+
+    strategy:
+      fail-fast: false
+      matrix:
+        os: [openstack22]
+
+    timeout-minutes: 600
+    steps:
+      - name: Checkout Rust source
+        uses: actions/checkout@v4
+        with:
+          submodules: true # check out all submodules so the cache can work correctly
+          fetch-depth: 2
+      - uses: dtolnay/rust-toolchain@nightly
+      - name: Create cache folders
+        run: |
+          mkdir -p build/build/x86_64-unknown-linux-gnu/llvm
+          mkdir -p build/build/x86_64-unknown-linux-gnu/enzyme
+          mkdir -p build/build/bootstrap/
+          mkdir -p build/build/x86_64-unknown-linux-gnu/stage0-rustc/
+          mkdir -p build/build/x86_64-unknown-linux-gnu/stage0-std/
+          mkdir -p build/build/x86_64-unknown-linux-gnu/stage0-tools/
+          mkdir -p build/build/x86_64-unknown-linux-gnu/stage1-std/
+      - name: Get LLVM commit hash
+        id: llvm-commit
+        run: echo "HEAD=$(git rev-parse HEAD:src/llvm-project)" >> $GITHUB_OUTPUT
+      - name: Cache LLVM
+        id: cache-llvm
+        uses: actions/cache@v4
+        with:
+          path: build/build/x86_64-unknown-linux-gnu/llvm
+          key: ${{ matrix.os }}-llvm-${{ steps.llvm-commit.outputs.HEAD }}
+      - name: Get Enzyme commit hash
+        id: enzyme-commit
+        run: echo "HEAD=$(git rev-parse HEAD:src/tools/enzyme)" >> $GITHUB_OUTPUT
+      - name: Cache Enzyme
+        id: cache-enzyme
+        uses: actions/cache@v4
+        with:
+          path: build/build/x86_64-unknown-linux-gnu/enzyme
+          key: ${{ matrix.os }}-enzyme-${{ steps.enzyme-commit.outputs.HEAD }}
+      - name: Cache bootstrap/stage0 artifacts for incremental builds
+        uses: actions/cache@v4
+        with:
+          path: |
+            build/build/bootstrap/
+            build/build/x86_64-unknown-linux-gnu/stage0-rustc/
+            build/build/x86_64-unknown-linux-gnu/stage0-std/
+            build/build/x86_64-unknown-linux-gnu/stage0-tools/
+            build/build/x86_64-unknown-linux-gnu/stage1-std/
+          # Approximate stable hash. It doesn't matter too much when this goes out of sync as it just caches
+          # some stage0/stage1 dependencies and stdlibs which *hopefully* are hash-keyed.
+          key: enzyme-rust-incremental-${{ runner.os }}-${{ hashFiles('src/**/Cargo.lock', 'Cargo.lock') }}
+          restore-keys: |
+            enzyme-rust-incremental-${{ runner.os }}
+      - name: Setup tmate session
+        uses: mxschmitt/action-tmate@v3
+      - name: Fail
+        run: 'false'
+          #      - name: Build
+          #        run: |
+          #          wget -O - https://apt.llvm.org/llvm-snapshot.gpg.key|sudo apt-key add -
+          #          sudo apt-add-repository "deb http://apt.llvm.org/`lsb_release -c | cut -f2`/ llvm-toolchain-`lsb_release -c | cut -f2`-17 main" || true
+          #          sudo apt-get -y update
+          #          sudo apt-get install -y lld-17
+          #          mkdir lld-path-manipulation
+          #          ln -s "$(which lld-17)" lld-path-manipulation/lld
+          #          ln -s "$(which lld-17)" lld-path-manipulation/ld
+          #          ln -s "$(which lld-17)" lld-path-manipulation/ld.lld
+          #          ln -s "$(which lld-17)" lld-path-manipulation/lld-17
+          #          export PATH="$PWD/lld-path-manipulation:$PATH"
+          #          mkdir -p build
+          #          cd build
+          #          rm -f config.toml
+          #          ../configure --enable-llvm-link-shared --enable-llvm-plugins --enable-llvm-enzyme --set=rust.use-lld=true --release-channel=nightly --enable-llvm-assertions --enable-option-checking --enable-ninja --disable-docs
+          #          ../x.py build --stage 1 library/std library/proc_macro library/test tools/rustdoc
+          #          rustup toolchain link enzyme build/host/stage1
+          #      - name: checkout Enzyme/rustbook
+          #        uses: actions/checkout@v4
+          #        with:
+          #          repository: EnzymeAD/rustbook
+          #          ref: main
+          #          path: rustbook
+          #      - name: test Enzyme/rustbook
+          #        working-directory: rustbook
+          #        run: |
+          #          cargo +enzyme test
+          #          ENZYME_LOOSE_TYPES=1 cargo +enzyme test -p samples-loose-types

.gitmodules

@@ -43,3 +43,7 @@
 	path = library/backtrace
 	url = https://github.com/rust-lang/backtrace-rs.git
 	shallow = true
+[submodule "src/tools/enzyme"]
+	path = src/tools/enzyme
+	url = https://github.com/enzymeAD/enzyme
+	shallow = true

Cargo.lock

@@ -4261,6 +4261,7 @@ dependencies = [
 name = "rustc_monomorphize"
 version = "0.0.0"
 dependencies = [
+ "rustc_ast",
  "rustc_data_structures",
  "rustc_errors",
  "rustc_fluent_macro",
@@ -4269,6 +4270,7 @@ dependencies = [
  "rustc_middle",
  "rustc_session",
  "rustc_span",
+ "rustc_symbol_mangling",
  "rustc_target",
  "serde",
  "serde_json",

README.md

@@ -1,9 +1,64 @@
-# The Rust Programming Language
+# The Rust Programming Language + AutoDiff
 
 [![Rust Community](https://img.shields.io/badge/Rust_Community%20-Join_us-brightgreen?style=plastic&logo=rust)](https://www.rust-lang.org/community)
 
 This is the main source code repository for [Rust]. It contains the compiler,
-standard library, and documentation.
+standard library, and documentation. It is modified to use Enzyme for AutoDiff.
+
+
+Please configure this fork using the following command:
+
+```
+mkdir build
+cd build
+../configure --enable-llvm-link-shared --enable-llvm-plugins --enable-llvm-enzyme --release-channel=nightly --enable-llvm-assertions --enable-clang --enable-lld --enable-option-checking --enable-ninja --disable-docs
+```
+
+Afterwards you can build rustc using:
+```
+../x.py build --stage 1 library
+```
+
+Afterwards rustc toolchain link will allow you to use it through cargo:
+```
+rustup toolchain link enzyme build/host/stage1
+rustup toolchain install nightly # enables -Z unstable-options
+```
+
+You can then run an examples from our [docs](https://enzyme.mit.edu/index.fcgi/rust/usage/usage.html) using
+
+```bash
+cargo +enzyme run --release
+```
+
+## Enzyme Config
+To help with debugging, Enzyme can be configured using environment variables.
+```bash
+export ENZYME_PRINT_TA=1
+export ENZYME_PRINT_AA=1
+export ENZYME_PRINT=1
+export ENZYME_PRINT_MOD=1
+export ENZYME_PRINT_MOD_AFTER=1
+```
+The first three will print TypeAnalysis, ActivityAnalysis and the llvm-ir on a function basis, respectively.
+The last two variables will print the whole module directly before and after Enzyme differented the functions. 
+
+When experimenting with flags please make sure that EnzymeStrictAliasing=0
+is not changed, since it is required for Enzyme to handle enums correctly.
+
+## Bug reporting
+Bugs are pretty much expected at this point of the development process.
+In order to help us please minimize the Rust code as far as possible.
+This tool might be a nicer helper: https://github.com/Nilstrieb/cargo-minimize
+If you have some knowledge of LLVM-IR we also greatly appreciate it if you could help
+us by compiling your minimized Rust code to LLVM-IR and reducing it further.
+
+The only exception to this strategy is error based on "Can not deduce type of X",
+where reducing your example will make it harder for us to understand the origin of the bug.
+In this case please just try to inline all dependencies into a single crate or even file,
+without deleting used code.
+
+
 
 [Rust]: https://www.rust-lang.org/
 

compiler/rustc_ast/src/ast.rs

@@ -2574,6 +2574,12 @@ impl FnRetTy {
             FnRetTy::Ty(ty) => ty.span,
         }
     }
+    pub fn has_ret(&self) -> bool {
+        match self {
+            FnRetTy::Default(_) => false,
+            FnRetTy::Ty(_) => true,
+        }
+    }
 }
 
 #[derive(Clone, Copy, PartialEq, Encodable, Decodable, Debug)]