TensorRT
TensorRT copied to clipboard
Add TORCHTRT_CHECK to execute_engine
Description
Convert some potential TensorRT errors (returned boolean is false) into proper exceptions, at runtime.
Fixes #2367
Type of change
- Bug fix (non-breaking change which fixes an issue)
Checklist:
- [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
- [ ] I have commented my code, particularly in hard-to-understand areas and hacks
- [ ] I have made corresponding changes to the documentation
- [ ] I have added tests to verify my fix or my feature
- [ ] New and existing unit tests pass locally with my changes
- [ ] I have added the relevant labels to my PR in so that relevant reviewers are notified
@peri044 can you look at this w.r.t TRT10?
Closing in favor of #2709