Skip to content
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

Dynamic Batch Support for TRT #6955

Merged
merged 34 commits into from
Dec 1, 2020

Conversation

codeislife99
Copy link
Contributor

This PR adds support for dynamic batch size for dense, conv , conv_transpose and ops surrounding them

@codeislife99 codeislife99 marked this pull request as draft November 23, 2020 18:22
@codeislife99 codeislife99 marked this pull request as ready for review November 24, 2020 20:37
@codeislife99 codeislife99 marked this pull request as draft November 24, 2020 20:39
@codeislife99
Copy link
Contributor Author

cc : @anijain2305 @trevor-m

@codeislife99 codeislife99 marked this pull request as ready for review November 24, 2020 22:44
Copy link
Contributor

@anijain2305 anijain2305 left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Overall LGTM. Please add comments. And I am not sure about the changes for CI.

python/tvm/relay/op/contrib/tensorrt.py Show resolved Hide resolved
python/tvm/relay/op/contrib/tensorrt.py Outdated Show resolved Hide resolved
tests/scripts/task_ci_python_setup.sh Outdated Show resolved Hide resolved
python/tvm/relay/op/contrib/tensorrt.py Outdated Show resolved Hide resolved
python/tvm/relay/op/contrib/tensorrt.py Show resolved Hide resolved
tests/python/contrib/test_tensorrt.py Outdated Show resolved Hide resolved
tests/python/contrib/test_tensorrt.py Show resolved Hide resolved
tests/python/contrib/test_tensorrt.py Outdated Show resolved Hide resolved
Copy link
Contributor

@trevor-m trevor-m left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thanks!

@anijain2305 anijain2305 merged commit 7950ea1 into apache:main Dec 1, 2020
@anijain2305
Copy link
Contributor

Thanks @codeislife99 @trevor-m This is merged!

trevor-m pushed a commit to trevor-m/tvm that referenced this pull request Dec 2, 2020
* add_annotate_fn

* Reshape_ann_fn

* Prune Subgraph

* Dynamic Shape

* Make PT Mask RCNN Work

* Cleanup

* Remove comments

* Remove COmments

* GetBatchSizeFix

* Fix Remove Droupout

* Fix Remove Droupout

* TRT Runtime

* Add MaskrCNN R50

* New Testing code

* Fix black

* Test Maskrcnn r50 done

* Test MR50

* Space typo

* Change Log to Dlog

* Move test to tensorrt.py

* Remove imports

* Remove function

* Add it to trt

* import error

* Imports

* Add torch to CI

* trt_test

* Check test

* Revert Pytorch install

* Fix

* test dynamic batch

* TRT

* Resolve PR comments

* Zero batch size add

Co-authored-by: Ubuntu <ubuntu@ip-172-31-27-149.us-east-2.compute.internal>
trevor-m pushed a commit to trevor-m/tvm that referenced this pull request Dec 4, 2020
* add_annotate_fn

* Reshape_ann_fn

* Prune Subgraph

* Dynamic Shape

* Make PT Mask RCNN Work

* Cleanup

* Remove comments

* Remove COmments

* GetBatchSizeFix

* Fix Remove Droupout

* Fix Remove Droupout

* TRT Runtime

* Add MaskrCNN R50

* New Testing code

* Fix black

* Test Maskrcnn r50 done

* Test MR50

* Space typo

* Change Log to Dlog

* Move test to tensorrt.py

* Remove imports

* Remove function

* Add it to trt

* import error

* Imports

* Add torch to CI

* trt_test

* Check test

* Revert Pytorch install

* Fix

* test dynamic batch

* TRT

* Resolve PR comments

* Zero batch size add

Co-authored-by: Ubuntu <ubuntu@ip-172-31-27-149.us-east-2.compute.internal>
trevor-m pushed a commit to neo-ai/tvm that referenced this pull request Dec 4, 2020
* add_annotate_fn

* Reshape_ann_fn

* Prune Subgraph

* Dynamic Shape

* Make PT Mask RCNN Work

* Cleanup

* Remove comments

* Remove COmments

* GetBatchSizeFix

* Fix Remove Droupout

* Fix Remove Droupout

* TRT Runtime

* Add MaskrCNN R50

* New Testing code

* Fix black

* Test Maskrcnn r50 done

* Test MR50

* Space typo

* Change Log to Dlog

* Move test to tensorrt.py

* Remove imports

* Remove function

* Add it to trt

* import error

* Imports

* Add torch to CI

* trt_test

* Check test

* Revert Pytorch install

* Fix

* test dynamic batch

* TRT

* Resolve PR comments

* Zero batch size add

Co-authored-by: Ubuntu <ubuntu@ip-172-31-27-149.us-east-2.compute.internal>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants