Skip to content

Commit 3a98a8b

Browse files
narendasannarendasan
authored
Merge pull request #691 from NVIDIA/torch_tensorrt_1.0
Torch-TensorRT 1.0.0
2 parents f99a6ca + 7ab76bc commit 3a98a8b

File tree

873 files changed

+152822
-57635
lines changed

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

873 files changed

+152822
-57635
lines changed

‎.bazelrc

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -33,4 +33,10 @@ build:python --define=target_lang=python
3333

3434
build:pre_cxx11_abi --cxxopt="-D_GLIBCXX_USE_CXX11_ABI=0"
3535
build:pre_cxx11_abi --linkopt="-D_GLIBCXX_USE_CXX11_ABI=0"
36-
build:pre_cxx11_abi --define=abi=pre_cxx11_abi
36+
build:pre_cxx11_abi --define=abi=pre_cxx11_abi
37+
38+
build:ci_testing --define=torchtrt_src=pre_built --cxxopt="-DDISABLE_TEST_IN_CI" --action_env "NVIDIA_TF32_OVERRIDE=0"
39+
build:use_precompiled_torchtrt --define=torchtrt_src=pre_built
40+
41+
test:ci_testing --define=torchtrt_src=pre_built --cxxopt="-DDISABLE_TEST_IN_CI" --action_env "NVIDIA_TF32_OVERRIDE=0"
42+
test:use_precompiled_torchtrt --define=torchtrt_src=pre_built

‎.bazelversion

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1 +1 @@
1-
4.0.0
1+
4.2.1

‎.dockerignore

Lines changed: 63 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,63 @@
1+
bazel-*
2+
*.tar.gz
3+
*.whl
4+
*.bin
5+
docs/v*/**/*
6+
env/
7+
bazel
8+
bazel-bazel-test
9+
bazel-bin
10+
bazel-genfiles
11+
bazel-out
12+
bazel-testlogs
13+
bazel-TRTorch
14+
bazel-trtorch-testing
15+
third_party/pytorch
16+
*.jit
17+
*.jit.pt
18+
.\#*
19+
experiments/
20+
py/build/
21+
py/tmp/
22+
py/.eggs
23+
.vscode/
24+
.DS_Store
25+
._DS_Store
26+
*.pth
27+
*.pyc
28+
cpp/ptq/training/vgg16/data/*
29+
*.bin
30+
cpp/ptq/datasets/data/
31+
tests/accuracy/datasets/data/*
32+
._.DS_Store
33+
*.tar.gz
34+
*.tgz
35+
docsrc/_build
36+
docsrc/_notebooks
37+
docsrc/_cpp_api
38+
docsrc/_tmp
39+
*.so
40+
__pycache__
41+
*.egg-info
42+
dist
43+
bdist
44+
py/torch_tensorrt/_version.py
45+
py/torch_tensorrt/lib
46+
py/torch_tensorrt/include
47+
py/torch_tensorrt/bin
48+
py/torch_tensorrt/BUILD
49+
py/torch_tensorrt/LICENSE
50+
py/torch_tensorrt/WORKSPACE
51+
py/wheelhouse
52+
py/.eggs
53+
notebooks/.ipynb_checkpoints/
54+
*.cache
55+
tests/py/data
56+
examples/**/deps/**/*
57+
examples/torchtrt_runtime_example/torchtrt_runtime_example
58+
examples/int8/ptq/ptq
59+
examples/int8/qat/qat
60+
examples/int8/training/vgg16/data/*
61+
examples/int8/datasets/data/*
62+
env/**/*
63+
*.ts

‎.github/ISSUE_TEMPLATE/--feature-request.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
---
22
name: "✨ Feature request"
33
about: Suggest an idea for this project
4-
title: "✨[Feature] An idea about a new feature in TRTorch"
4+
title: "✨[Feature] An idea about a new feature in Torch-TensorRT"
55
labels: feature request
66
assignees: narendasan
77

‎.github/ISSUE_TEMPLATE/--question.md

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
---
22
name: "❓ Question"
3-
about: Ask a question about using TRTorch
3+
about: Ask a question about using Torch-TensorRT
44
title: "❓ [Question] How do you ....? "
55
labels: question
66
assignees: ''
@@ -17,7 +17,7 @@ assignees: ''
1717

1818
## Environment
1919

20-
> Build information about the TRTorch compiler can be found by turning on debug messages
20+
> Build information about Torch-TensorRT can be found by turning on debug messages
2121
2222
- PyTorch Version (e.g., 1.0):
2323
- CPU Architecture:

‎.github/ISSUE_TEMPLATE/bug-report.md

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
---
22
name: "\U0001F41B Bug Report"
3-
about: Submit a bug report on an issue encountered with TRTorch
4-
title: "\U0001F41B [Bug] Encountered bug when using TRTorch"
3+
about: Submit a bug report on an issue encountered with Torch-TensorRT
4+
title: "\U0001F41B [Bug] Encountered bug when using Torch-TensorRT"
55
labels: bug
66
assignees: ''
77

@@ -27,9 +27,9 @@ Steps to reproduce the behavior:
2727

2828
## Environment
2929

30-
> Build information about the TRTorch compiler can be found by turning on debug messages
30+
> Build information about Torch-TensorRT can be found by turning on debug messages
3131
32-
- TRTorch Version (e.g. 0.2.0):
32+
- Torch-TensorRT Version (e.g. 1.0.0):
3333
- PyTorch Version (e.g. 1.0):
3434
- CPU Architecture:
3535
- OS (e.g., Linux):

‎.github/ISSUE_TEMPLATE/op-converter-request.md

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
---
22
name: "↔ Op Converter Request"
3-
about: Submit a proposal/request to support a new PyTorch operator in TRTorch
4-
title: "↔ [Converter] Add support for my_op in TRTorch"
3+
about: Submit a proposal/request to support a new PyTorch operator in Torch-TensorRT
4+
title: "↔ [Converter] Add support for my_op in Torch-TensorRT"
55
labels: 'component: converters, feature request'
66
assignees: ''
77

‎.github/pr-labels.yml

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
1-
"component: trtorchc":
2-
- cpp/trtorchc/**/*
3-
1+
"component: torchtrtc":
2+
- cpp/torchtrtc/**/*
3+
44
"component: api [C++]":
55
- cpp/**/*
66

‎.github/workflows/docgen.yml

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@ jobs:
1212
build-docs:
1313
runs-on: ubuntu-18.04
1414
container:
15-
image: docker.pkg.github.com/nvidia/trtorch/docgen:latest
15+
image: docker.pkg.github.com/nvidia/torch-tensorrt/docgen:latest
1616
credentials:
1717
username: $GITHUB_ACTOR
1818
password: ${{secrets.GITHUB_TOKEN}}
@@ -36,14 +36,14 @@ jobs:
3636
- name: Generate New Docs
3737
run: |
3838
cd docsrc
39-
python3 -c "import trtorch; print(trtorch.__version__)"
39+
python3 -c "import torch_tensorrt; print(torch_tensorrt.__version__)"
4040
make html
4141
- uses: stefanzweifel/git-auto-commit-action@v4
4242
with:
4343
# Required
4444
commit_message: "docs: [Automated] Regenerating documenation for ${{ steps.vars.outputs.sha }}"
4545
commit_options: "--no-verify --signoff"
4646
file_pattern: docs/
47-
commit_user_name: TRTorch Github Bot
48-
commit_user_email: trtorch[email protected]
49-
commit_author: TRTorch Github Bot <trtorch[email protected]>
47+
commit_user_name: Torch-TensorRT Github Bot
48+
commit_user_email: torch-tensorrt[email protected]
49+
commit_author: Torch-TensorRT Github Bot <torch-tensorrt[email protected]>

‎.github/workflows/linter.yml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,7 @@ jobs:
1818
GITHUB_TOKEN: ${{secrets.GITHUB_TOKEN}}
1919
- name: Run image
2020
run: |
21-
docker run -it -d --name cpplinter -e GITHUB_TOKEN=$GITHUB_TOKEN -v $GITHUB_WORKSPACE:/workspace -v $GITHUB_EVENT_PATH:/GITHUB_EVENT.json -w /workspace docker.pkg.github.com/nvidia/trtorch/docgen:latest
21+
docker run -it -d --name cpplinter -e GITHUB_TOKEN=$GITHUB_TOKEN -v $GITHUB_WORKSPACE:/workspace -v $GITHUB_EVENT_PATH:/GITHUB_EVENT.json -w /workspace docker.pkg.github.com/nvidia/torch-tensorrt/docgen:latest
2222
docker exec cpplinter bash -c "cp /workspace/docker/WORKSPACE.docs /workspace/WORKSPACE"
2323
env:
2424
GITHUB_TOKEN: ${{secrets.GITHUB_TOKEN}}
@@ -40,7 +40,7 @@ jobs:
4040
GITHUB_TOKEN: ${{secrets.GITHUB_TOKEN}}
4141
- name: Run image
4242
run: |
43-
docker run -it -d --name pylinter -e GITHUB_TOKEN=$GITHUB_TOKEN -v $GITHUB_WORKSPACE:/workspace -v $GITHUB_EVENT_PATH:/GITHUB_EVENT.json -w /workspace docker.pkg.github.com/nvidia/trtorch/docgen:latest
43+
docker run -it -d --name pylinter -e GITHUB_TOKEN=$GITHUB_TOKEN -v $GITHUB_WORKSPACE:/workspace -v $GITHUB_EVENT_PATH:/GITHUB_EVENT.json -w /workspace docker.pkg.github.com/nvidia/torch-tensorrt/docgen:latest
4444
docker exec pylinter bash -c "cp /workspace/docker/WORKSPACE.docs /workspace/WORKSPACE"
4545
env:
4646
GITHUB_TOKEN: ${{secrets.GITHUB_TOKEN}}

0 commit comments

Comments
 (0)