TensorRT
TensorRT copied to clipboard
feat(//tools/perf): Refactor perf_run.py, add fx2trt backend support, usage via CLI arguments
Description
Refactor perf_run.py, add fx2trt backend support, usage via CLI arguments. Added a benchmark.sh which is used for internal perf regression testing.
Type of change
- New feature (non-breaking change which adds functionality)
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
- [x] 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
- [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
I think this branch needs to be rebased seems like the pre-commit was not run
Merging this PR as this doesn't affect the library. Please re-open incase you see any issues