fix: Allow rank differences in aten.expand
(#2234)
#816
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: docker.io/pytorch/manylinux-builder:cuda12.1 | |
steps: | |
- uses: actions/checkout@v3 | |
with: | |
ref: ${{github.head_ref}} | |
- name: Install base deps | |
run: | | |
./packaging/pre_build_script.sh | |
- 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: Build Python Package | |
run: | | |
cp toolchains/ci_workspaces/WORKSPACE.x86_64.cu121.release.rhel WORKSPACE | |
python -m pip install pip<=23 | |
python -m pip install --pre -e . --extra-index-url https://download.pytorch.org/whl/nightly/cu121 | |
- name: Generate New Docs | |
run: | | |
cd docsrc | |
python -m pip install -r requirements.txt | |
python -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]> |