-
Notifications
You must be signed in to change notification settings - Fork 3.6k
Add dynamic shape support in TensorRT execution provider #2450
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Merged
Merged
Changes from all commits
Commits
Show all changes
85 commits
Select commit
Hold shift + click to select a range
2ec7ddb
remove onnx-tensorrt submodule
stevenlix d4eea2f
add new onnx-tensorrt submodule (experiment) for trt6
stevenlix d50c974
update engine build for trt6
stevenlix 3b89475
update compile and compute for tensorrt6.0
stevenlix ef51e18
Update tensorrt_execution_provider.cc
stevenlix ae07c80
Update tensorrt_execution_provider.cc
stevenlix 2f043de
Update tensorrt_execution_provider.cc
stevenlix 14d31cd
Update tensorrt_execution_provider.cc
stevenlix 7a47cac
switch to onnx-tensorrt master for TensorRT6'
stevenlix 2abe72b
Merge branch 'stevenlix/trt6' of https://github.com/Microsoft/onnxrun…
stevenlix 902181e
Update tensorrt_execution_provider.cc
stevenlix 694fb9a
Handle dynamic batch size and add memcpy in TensorRT EP
stevenlix ef23066
update test cases
stevenlix 84dad1a
Update tensorrt_execution_provider.cc
stevenlix ed2693e
update onnx-tensorrt submodule
stevenlix 1febce8
Merge branch 'stevenlix/trt6' of https://github.com/Microsoft/onnxrun…
stevenlix c243f53
Update Dockerfile.ubuntu_tensorrt
stevenlix 2091b68
merge master
stevenlix 9ca8d38
Update Dockerfile.ubuntu_tensorrt
stevenlix 70e905d
Update run_dockerbuild.sh
stevenlix 29ade9c
Update run_dockerbuild.sh
stevenlix 2050cb8
Update install_ubuntu.sh
stevenlix 6ef7cf2
Update concat_op_test.cc
stevenlix 37cdc7d
Update tensorrt_execution_provider.cc
stevenlix cc21556
Upgrade TensorRT to version 6.0.1.5
stevenlix 88a4b7a
Merge branch 'stevenlix/trt6' of https://github.com/Microsoft/onnxrun…
stevenlix 76ad980
Update onnxruntime_providers.cmake
stevenlix cfba487
Update CMakeLists.txt
stevenlix c4316d3
Update reduction_ops_test.cc
stevenlix 1a43bb3
Merge branch 'master' into stevenlix/trt6
stevenlix 27487d4
Update install_ubuntu.sh
stevenlix f7ccd5a
Update Dockerfile.ubuntu_tensorrt
stevenlix 0f3dc0a
Update Dockerfile.tensorrt
stevenlix c427ede
Update BUILD.md
stevenlix 45781ff
Update run_dockerbuild.sh
stevenlix ddde6f6
Update install_ubuntu.sh
stevenlix 3b5f551
Update onnxruntime_providers.cmake
stevenlix 909765a
Update install_ubuntu.sh
stevenlix 8f2c7b3
Update install_ubuntu.sh
stevenlix 94c4887
Update gemm_test.cc
stevenlix b54d9e9
Update gather_op_test.cc
stevenlix 18dfe23
Update CMakeLists.txt
stevenlix ab37d25
Removed submodule
stevenlix c315927
update onnx-tensorrt submodule
stevenlix 08ab724
Merge remote-tracking branch 'origin/stevenlix/trt6' into stevenlix/d…
stevenlix 63eff0c
update header file
stevenlix 9f455a8
resolve conflict
stevenlix 681cc57
Removed submodule
stevenlix 7ebf729
add submodule onnx-tensorrt kevin's branch shape-test'
stevenlix 7b09e3f
add debugging code
stevenlix a9e3e96
Update tensorrt_execution_provider.cc
stevenlix b165e63
Update tensorrt_execution_provider.cc
stevenlix b185773
merge master
stevenlix 580940d
merge master
stevenlix aac11b4
Removed submodule
stevenlix 51b07f9
update onnx-tensorrt submodule
stevenlix 9a24ea0
add more changes for dynamic shapes
stevenlix 837c8f4
Update tensorrt_execution_provider.cc
stevenlix 7777308
update for dynamic shape
35460f7
Merge branch 'master' into stevenlix/dynamicshape
a6eb1d1
update dynamic shape processing
d046f57
Merge branch 'master' into stevenlix/dynamicshape
6d2b372
fix logger issue
153b043
Merge remote-tracking branch 'origin/master' into stevenlix/dynamicshape
c098bea
remove submodule onnx-tensorrt
stevenlix 02c4ab4
add submodule onnx-tensorrt
stevenlix e2e40fc
add env variable min_subgraph_size
stevenlix bf2a6e8
remove redundency
stevenlix f1c9374
update document
a571e37
use onnxruntime::make_unique
stevenlix 62ffa6e
solve conflict
001c0f4
fix multi-run issue
8740a2e
remove some tests to save CI build time
ef4ecb7
Add dynamic shape test
stevenlix 1673803
Update TensorRT-ExecutionProvider.md
stevenlix 5494497
Add example of running Faster R-CNN model on TensorRT EP
stevenlix 5c7e8f4
Add more details on env variables
stevenlix 1759868
update environment variables
stevenlix 2b6a974
Update tensorrt_basic_test.cc
stevenlix ac70ec8
Merge branch 'master' into stevenlix/dynamicshape
stevenlix e63214f
Update model tests
stevenlix e41ff98
Update tensor_op_test.cc
stevenlix 30240e8
remove --use_full_protobuf
stevenlix 9564b9f
Update build.py
stevenlix e7fa3f7
Merge branch 'master' into stevenlix/dynamicshape
jywu-msft File filter
Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
There are no files selected for viewing
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
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
Submodule onnx-tensorrt
updated
17 files
| +1 −0 | .gitmodules | |
| +2 −3 | CMakeLists.txt | |
| +0 −91 | Dockerfile | |
| +21 −1 | ImporterContext.hpp | |
| +66 −80 | ModelImporter.cpp | |
| +1 −0 | ModelImporter.hpp | |
| +2 −2 | NvOnnxParser.h | |
| +16 −3 | README.md | |
| +3 −2 | ShapedWeights.cpp | |
| +422 −624 | builtin_op_importers.cpp | |
| +70 −0 | docker/onnx-tensorrt-deb.Dockerfile | |
| +80 −0 | docker/onnx-tensorrt-tar.Dockerfile | |
| +496 −66 | onnx2trt_utils.cpp | |
| +58 −4 | onnx2trt_utils.hpp | |
| +7 −1 | onnx_utils.hpp | |
| +1 −1 | setup.py | |
| +1 −1 | third_party/onnx |
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
Oops, something went wrong.
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
it's better to do a notebook tutorial to be consistent with other EPs.
let's do it as a separate future PR.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Will do