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

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

- 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

April 20, 2023 18:40

bowang007 pushed a commit that referenced this pull request

Apr 28, 2023