chore: dynamic shape support for flip ops (#3046) #1530
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
| # This is a basic workflow to help you get started with Actions | |
| name: Generate Docs | |
| # Controls when the action will run. Triggers the workflow on push or pull request | |
| # events but only for the master branch | |
| on: | |
| push: | |
| branches: [ main ] | |
| jobs: | |
| build-docs: | |
| runs-on: linux.g5.4xlarge.nvidia.gpu | |
| if: ${{ ! contains(github.actor, 'pytorchbot') }} | |
| environment: pytorchbot-env | |
| container: | |
| image: docker.io/pytorch/manylinux-builder:cuda12.4 | |
| options: --gpus all | |
| env: | |
| CUDA_HOME: /usr/local/cuda-12.4 | |
| VERSION_SUFFIX: cu124 | |
| CU_VERSION: cu124 | |
| CHANNEL: nightly | |
| CI_BUILD: 1 | |
| steps: | |
| - uses: actions/checkout@v3 | |
| with: | |
| ref: ${{github.head_ref}} | |
| token: ${{ secrets.GH_PYTORCHBOT_TOKEN }} | |
| - name: Select Python / CUDA | |
| run: | | |
| git config --global --add safe.directory /__w/TensorRT/TensorRT | |
| echo "/opt/python/cp311-cp311/bin/" >> $GITHUB_PATH | |
| - name: Install base deps | |
| run: | | |
| python3 -m pip install pip --upgrade | |
| python3 -m pip install pyyaml numpy torch --pre --extra-index-url https://download.pytorch.org/whl/nightly/cu124 | |
| ./packaging/pre_build_script.sh | |
| - name: Get HEAD SHA | |
| id: vars | |
| run: echo "sha=$(git rev-parse --short HEAD)" >> $GITHUB_OUTPUT | |
| - name: Build Python Package | |
| run: | | |
| python3 -m pip install --pre . --extra-index-url https://download.pytorch.org/whl/nightly/cu124 | |
| - name: Generate New Docs | |
| run: | | |
| cd docsrc | |
| yum install -y doxygen pandoc | |
| python3 -m pip install -r requirements.txt | |
| python3 -c "import torch_tensorrt; print(torch_tensorrt.__version__)" | |
| make html | |
| cd .. | |
| - uses: stefanzweifel/git-auto-commit-action@v4 | |
| with: | |
| # Required | |
| commit_message: "docs: [Automated] Regenerating documenation for ${{ steps.vars.outputs.sha }}" | |
| commit_options: "--no-verify --signoff" | |
| file_pattern: docs/ | |
| commit_user_name: Torch-TensorRT Github Bot | |
| commit_user_email: [email protected] | |
| commit_author: Torch-TensorRT Github Bot <[email protected]> | |
| concurrency: | |
| group: ${{ github.workflow }}-${{ github.ref_name }} | |
| cancel-in-progress: true |