fix: Error caused by invalid binding name in `TRTEngine.to_str()` method by gs-olive · Pull Request #1846 · pytorch/TensorRT (original) (raw)

Description

Fix to_str() method using invalid binding names for TRT Engines
Method was using standard I/O names of the form input_*, output_*, which does not represent the full range of potential names
Update implementation to access binding names directly from binding names array
Update debug message printing to improve readability + clarity + provided information

Fixes #1844

[ x ] My code follows the style guidelines of this project (You can use the linters)
[ x ] I have performed a self-review of my own code
[ x ] I have commented my code, particularly in hard-to-understand areas and hacks
[ x ] I have made corresponding changes to the documentation
[ x ] I have added tests to verify my fix or my feature
[ x ] New and existing unit tests pass locally with my changes
[ x ] I have added the relevant labels to my PR in so that relevant reviewers are notified

Fix to_str() method using invalid binding names for TRT Engines
Method was using standard I/O names of the form input_*, output_*, which does not represent the full range of potential names
Update implementation to access binding names directly from binding names array
Update debug message printing to improve readability + clarity

gs-olive marked this pull request as ready for review

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

LGTM

bowang007 pushed a commit that referenced this pull request