[mlir-tensorrt] Transpose Reshape Elimination pass #846
Workflow file for this run
This file contains hidden or 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: MLIR-TensorRT CI | |
| on: | |
| pull_request: | |
| branches: | |
| - main | |
| types: [synchronize, opened, reopened, ready_for_review] | |
| paths: ["mlir-tensorrt/**"] | |
| push: | |
| branches: | |
| - main | |
| paths: ["mlir-tensorrt/**"] | |
| env: | |
| DEFAULT_IMAGE: ghcr.io/nvidia/tensorrt-incubator/mlir-tensorrt:cuda12.5-ubuntu-llvm17 | |
| REGISTRY: ghcr.io | |
| jobs: | |
| mlir-tensorrt-test-pr: | |
| if: github.event_name != 'pull_request' || github.event.pull_request.draft == false | |
| runs-on: tripy-self-hosted | |
| steps: | |
| # Free some disk space, otherwise we get OOM error. | |
| - name: Free disk space | |
| run: | | |
| sudo rm -rf \ | |
| /usr/share/dotnet "$AGENT_TOOLSDIRECTORY" /usr/local/lib/android /opt/ghc \ | |
| /usr/local/share/powershell /usr/share/swift /usr/local/.ghcup \ | |
| /usr/lib/jvm | |
| sudo apt-get purge microsoft-edge-stable || true | |
| sudo apt-get purge google-cloud-cli || true | |
| sudo apt-get purge dotnet-sdk-* || true | |
| sudo apt-get purge google-chrome-stable || true | |
| sudo apt-get autoremove -y | |
| sudo apt-get autoclean -y | |
| # Value of `github.workspace` is /home/runner/work/{repo_name}/{repo-name} | |
| # i.e. /home/runner/work/TensorRT-Incubator/TensorRT-Incubator in our case. | |
| # After this action, repo is cloned inside above path. | |
| - uses: actions/checkout@v4 | |
| with: | |
| fetch-depth: 5 | |
| # Run initial format check | |
| - name: Run python format and clang check | |
| uses: addnab/docker-run-action@v3 | |
| if: ${{ github.event_name == 'pull_request' }} | |
| with: | |
| image: ${{ env.DEFAULT_IMAGE }} | |
| options: -v ${{ github.workspace }}:/tensorrt-incubator | |
| registry: ${{ env.REGISTRY }} | |
| username: ${{ github.actor }} | |
| password: ${{ secrets.GITHUB_TOKEN }} | |
| # This step does two things | |
| # 1. Check if Python files follow black format | |
| # 2. Check if C++ files follow clang format | |
| # NOTE: We are placed at the root directory ('/') inside the container. | |
| run: | | |
| cd tensorrt-incubator | |
| git config --global --add safe.directory /tensorrt-incubator | |
| cat > run_format_check.sh <<EOF | |
| #!/bin/bash | |
| set -e | |
| python3 -m black --check --extend-exclude='.*\.pyi' mlir-tensorrt/compiler/ | |
| python3 -m black --check --extend-exclude='.*\.pyi' mlir-tensorrt/integrations/python/ | |
| git clang-format HEAD~1 --diff | |
| EOF | |
| bash run_format_check.sh | |
| # Create cache folders | |
| - name: Create cache folder | |
| run: | | |
| mkdir -p ${{ github.workspace }}/ccache | |
| mkdir -p ${{ github.workspace }}/.cache.cpm | |
| # Restore cache, if exists. | |
| - name: Restore cache | |
| id: restore-cache | |
| uses: actions/cache/restore@v4 | |
| with: | |
| key: ${{ runner.os }}-mlir-tensorrt-cache-${{ hashFiles('mlir-tensorrt/**/*.cpp', 'mlir-tensorrt/**/*.h', 'mlir-tensorrt/build_tools/**/*') }} | |
| restore-keys: | | |
| ${{ runner.os }}-mlir-tensorrt-cache- | |
| path: | | |
| ${{ github.workspace }}/ccache | |
| ${{ github.workspace }}/.cache.cpm/* | |
| !${{ github.workspace }}/.cache.cpm/tensorrt | |
| # TensorRT 10 tests | |
| - name: TensorRT 10 build | |
| uses: addnab/docker-run-action@v3 | |
| with: | |
| image: ${{ env.DEFAULT_IMAGE }} | |
| options: -v ${{ github.workspace }}/mlir-tensorrt:/mlir-tensorrt -v ${{ github.workspace }}/ccache:/ccache -v ${{ github.workspace }}/.cache.cpm:/.cache.cpm --gpus all | |
| registry: ${{ env.REGISTRY }} | |
| username: ${{ github.actor }} | |
| password: ${{ secrets.GITHUB_TOKEN }} | |
| run: | | |
| cd mlir-tensorrt | |
| ./build_tools/scripts/cicd_build.sh --build_only | |
| - name: Save cache | |
| id: save-cache | |
| uses: actions/cache/save@v4 | |
| with: | |
| key: ${{ runner.os }}-mlir-tensorrt-cache-${{ hashFiles('mlir-tensorrt/**/*.cpp', 'mlir-tensorrt/**/*.h', 'mlir-tensorrt/build_tools/**/*') }} | |
| path: | | |
| ${{ github.workspace }}/ccache | |
| ${{ github.workspace }}/.cache.cpm/* | |
| !${{ github.workspace }}/.cache.cpm/tensorrt | |
| - name: TensorRT 10 test | |
| uses: addnab/docker-run-action@v3 | |
| with: | |
| image: ${{ env.DEFAULT_IMAGE }} | |
| options: -v ${{ github.workspace }}/mlir-tensorrt:/mlir-tensorrt -v ${{ github.workspace }}/ccache:/ccache -v ${{ github.workspace }}/.cache.cpm:/.cache.cpm --gpus all | |
| registry: ${{ env.REGISTRY }} | |
| username: ${{ github.actor }} | |
| password: ${{ secrets.GITHUB_TOKEN }} | |
| run: | | |
| cd mlir-tensorrt | |
| ./build_tools/scripts/cicd_build.sh | |
| # TensorRT 10 & ASAN | |
| - name: TensorRT 10 ASAN test | |
| uses: addnab/docker-run-action@v3 | |
| with: | |
| image: ${{ env.DEFAULT_IMAGE }} | |
| options: -v ${{ github.workspace }}/mlir-tensorrt:/mlir-tensorrt -v ${{ github.workspace }}/ccache:/ccache -v ${{ github.workspace }}/.cache.cpm:/.cache.cpm --gpus all | |
| registry: ${{ env.REGISTRY }} | |
| username: ${{ github.actor }} | |
| password: ${{ secrets.GITHUB_TOKEN }} | |
| run: | | |
| cd mlir-tensorrt | |
| ENABLE_ASAN=ON ./build_tools/scripts/cicd_build.sh |