mlir: implement MemorySlot Interfaces for Gradient ops #7801
Workflow file for this run
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| name: Benchmarking | |
| on: | |
| push: | |
| branches: | |
| - main | |
| pull_request: | |
| branches: | |
| - main | |
| merge_group: | |
| concurrency: | |
| group: ${{ github.workflow }}-${{ github.head_ref || github.run_id }} | |
| cancel-in-progress: true | |
| jobs: | |
| build: | |
| name: Benchmark Test on os ${{ matrix.os }} and llvm ${{ matrix.llvm }} mode ${{ matrix.build }} | |
| runs-on: ${{ matrix.os }} | |
| strategy: | |
| fail-fast: false | |
| matrix: | |
| llvm: ["16", "17", "18"] | |
| build: ["Release"] #, "Debug" "RelWithDebInfo" | |
| os: [openstack22] | |
| timeout-minutes: 120 | |
| steps: | |
| - name: add llvm | |
| 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`-${{ matrix.llvm }} main" || true | |
| sudo apt-get install -y python3-pip autoconf cmake gcc g++ libtool gfortran libblas-dev llvm-${{ matrix.llvm }}-dev clang-${{ matrix.llvm }} libeigen3-dev libboost-dev libzstd-dev | |
| sudo python3 -m pip install lit pathlib | |
| sudo touch /usr/lib/llvm-${{ matrix.llvm }}/bin/yaml-bench | |
| - uses: actions/checkout@v4 | |
| - name: mkdir | |
| run: rm -rf build && mkdir build | |
| - name: cmake | |
| working-directory: build | |
| run: cmake ../enzyme -DLLVM_EXTERNAL_LIT=`which lit` -DCMAKE_BUILD_TYPE=${{ matrix.build }} -DLLVM_DIR=/usr/lib/llvm-${{ matrix.llvm }}/lib/cmake/llvm | |
| - name: make | |
| working-directory: build | |
| run: make -j `nproc` LLVMEnzyme-${{ matrix.llvm }} | |
| - name: make bench-enzyme | |
| working-directory: build | |
| run: make bench-enzyme | |
| - name: results | |
| run: cat enzyme/benchmarks/*/*/results.txt | |
| - name: graph results | |
| if: github.event_name != 'pull_request' && matrix.build == 'Release' | |
| run: | | |
| for p in enzyme/benchmarks/ReverseMode/*/results.json; do | |
| python3 enzyme/benchmarks/upload-results.py $p --url https://enzyme.mit.edu/cibench/api --token ${{ secrets.GRAPH_TOKEN }} | |
| done |