fix: TRTInterpreter output lacks return value (#2114) #776
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: ubuntu-20.04 | |
| container: | |
| image: ghcr.io/pytorch/tensorrt/docgen:latest | |
| credentials: | |
| username: ${{ github.actor }} | |
| password: ${{ secrets.GITHUB_TOKEN }} | |
| steps: | |
| - name: Reclaim space | |
| run: | | |
| rm -rf /usr/share/dotnet | |
| rm -rf /opt/ghc | |
| rm -rf "/usr/local/share/boost" | |
| rm -rf /usr/local/cuda/cuda-* | |
| - name: Install base deps | |
| run: | | |
| apt update | |
| DEBIAN_FRONTEND=noninteractive apt install -y software-properties-common gcc git curl wget make zlib1g-dev bzip2 libbz2-dev lzma lzma-dev libreadline-dev libsqlite3-dev libssl-dev libffi-dev doxygen pandoc | |
| git config --global --add safe.directory '*' | |
| - name: Set up Python 3.10.12 | |
| uses: actions/setup-python@v4 | |
| with: | |
| python-version: 3.10.12 | |
| - uses: actions/checkout@v3 | |
| with: | |
| ref: ${{github.head_ref}} | |
| - name: Get HEAD SHA | |
| id: vars | |
| run: echo "sha=$(git rev-parse --short HEAD)" >> $GITHUB_OUTPUT | |
| - name: Get Bazel version | |
| id: bazel_info | |
| run: echo "version=$(cat .bazelversion)" >> $GITHUB_OUTPUT | |
| - name: Install Bazel | |
| run: | | |
| wget -q https://github.com/bazelbuild/bazel/releases/download/${{ steps.bazel_info.outputs.version }}/bazel-${{ steps.bazel_info.outputs.version }}-linux-x86_64 -O /usr/bin/bazel | |
| chmod a+x /usr/bin/bazel | |
| - name: Install cudnn + tensorrt | |
| run: | | |
| apt-get update | |
| wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-ubuntu2004.pin | |
| mv cuda-ubuntu2004.pin /etc/apt/preferences.d/cuda-repository-pin-600 | |
| apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/7fa2af80.pub | |
| apt-key adv --keyserver keyserver.ubuntu.com --recv-keys 536F8F1DE80F6A35 | |
| apt-key adv --keyserver keyserver.ubuntu.com --recv-keys A4B469963BF863CC | |
| add-apt-repository "deb https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/ /" | |
| apt-get update | |
| apt-get install -y libcudnn8 libcudnn8-dev | |
| apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/3bf863cc.pub | |
| add-apt-repository "deb https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/ /" | |
| apt-get update | |
| apt-get install -y libnvinfer8 libnvinfer-plugin8 libnvinfer-dev libnvinfer-plugin-dev | |
| - name: Install Torch | |
| run: | | |
| python3 -m pip install -r py/requirements.txt --user | |
| - name: Build Python Package | |
| run: | | |
| cp toolchains/ci_workspaces/WORKSPACE.x86_64 WORKSPACE | |
| cd py | |
| python3 -m pip install pip==21.3.1 | |
| echo $(which python3) | |
| echo $(python3 -c 'import site; print(site.getsitepackages()[0])') | |
| mkdir -p /opt/circleci/.pyenv/versions/3.9.4/lib/python3.9/ | |
| ln -s $(python3 -c 'import site; print(site.getsitepackages()[0])') /opt/circleci/.pyenv/versions/3.9.4/lib/python3.9/site-packages | |
| python3 setup.py install | |
| cd .. | |
| - name: Generate New Docs | |
| run: | | |
| cd docsrc | |
| 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: torch-tensorrt.github.bot@nvidia.com | |
| commit_author: Torch-TensorRT Github Bot <torch-tensorrt.github.bot@nvidia.com> |