-
Notifications
You must be signed in to change notification settings - Fork 763
feat(autoware_tensorrt_plugins): implemented a tensorrt plugin for the argsort operation #10601
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
feat(autoware_tensorrt_plugins): implemented a tensorrt plugin for the argsort operation #10601
Conversation
Signed-off-by: Kenzo Lobos-Tsunekawa <[email protected]>
Signed-off-by: Kenzo Lobos-Tsunekawa <[email protected]>
Thank you for contributing to the Autoware project! 🚧 If your pull request is in progress, switch it to draft mode. Please ensure:
|
Signed-off-by: Kenzo Lobos-Tsunekawa <[email protected]>
Signed-off-by: Kenzo Lobos-Tsunekawa <[email protected]>
Signed-off-by: Kenzo Lobos-Tsunekawa <[email protected]>
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Only "style" comments.
LGTM!
perception/autoware_tensorrt_plugins/src/argsort_ops/argsort.cu
Outdated
Show resolved
Hide resolved
perception/autoware_tensorrt_plugins/src/argsort_plugin_creator.cpp
Outdated
Show resolved
Hide resolved
Signed-off-by: Kenzo Lobos-Tsunekawa <[email protected]>
Signed-off-by: Kenzo Lobos-Tsunekawa <[email protected]>
Signed-off-by: Kenzo Lobos-Tsunekawa <[email protected]>
Signed-off-by: Kenzo Lobos-Tsunekawa <[email protected]>
@amadeuszsz |
@knzo25 We need to wait for deployment - this workflow success. |
Signed-off-by: Kenzo Lobos-Tsunekawa <[email protected]>
…oware.universe into feat/tensorrt_plugins_argsort
Codecov ReportAttention: Patch coverage is
Additional details and impacted files@@ Coverage Diff @@
## main #10601 +/- ##
==========================================
- Coverage 24.85% 24.67% -0.18%
==========================================
Files 1351 1371 +20
Lines 105078 105802 +724
Branches 39603 39632 +29
==========================================
- Hits 26121 26111 -10
- Misses 76473 77207 +734
Partials 2484 2484
*This pull request uses carry forward flags. Click here to find out more. ☔ View full report in Codecov by Sentry. 🚀 New features to boost your workflow:
|
Signed-off-by: Kenzo Lobos-Tsunekawa <[email protected]>
Signed-off-by: Kenzo Lobos-Tsunekawa <[email protected]>
Signed-off-by: Kenzo Lobos-Tsunekawa <[email protected]>
…e argsort operation (autowarefoundation#10601) * feat: implemented a argsort plugin for tensorrt Signed-off-by: Kenzo Lobos-Tsunekawa <[email protected]> * fix: forgot to add the plugin to the initalizer list Signed-off-by: Kenzo Lobos-Tsunekawa <[email protected]> * chore: cspell Signed-off-by: Kenzo Lobos-Tsunekawa <[email protected]> * fix: mistook the compute capabilities of edge devices Signed-off-by: Kenzo Lobos-Tsunekawa <[email protected]> * chore: added tensorrt 10 or later as a requirement Signed-off-by: Kenzo Lobos-Tsunekawa <[email protected]> * chore: replaced 0 by nullptr Signed-off-by: Kenzo Lobos-Tsunekawa <[email protected]> * chore: removed unused headers Signed-off-by: Kenzo Lobos-Tsunekawa <[email protected]> * chore: removed double slash Signed-off-by: Kenzo Lobos-Tsunekawa <[email protected]> * chore: removed unused headers Signed-off-by: Kenzo Lobos-Tsunekawa <[email protected]> * chore: ci/cd Signed-off-by: Kenzo Lobos-Tsunekawa <[email protected]> * chore: ci/cd Signed-off-by: Kenzo Lobos-Tsunekawa <[email protected]> * fix: forgot the noescept in the definition Signed-off-by: Kenzo Lobos-Tsunekawa <[email protected]> * chore: ci/cd Signed-off-by: Kenzo Lobos-Tsunekawa <[email protected]> --------- Signed-off-by: Kenzo Lobos-Tsunekawa <[email protected]>
…e argsort operation (autowarefoundation#10601) * feat: implemented a argsort plugin for tensorrt Signed-off-by: Kenzo Lobos-Tsunekawa <[email protected]> * fix: forgot to add the plugin to the initalizer list Signed-off-by: Kenzo Lobos-Tsunekawa <[email protected]> * chore: cspell Signed-off-by: Kenzo Lobos-Tsunekawa <[email protected]> * fix: mistook the compute capabilities of edge devices Signed-off-by: Kenzo Lobos-Tsunekawa <[email protected]> * chore: added tensorrt 10 or later as a requirement Signed-off-by: Kenzo Lobos-Tsunekawa <[email protected]> * chore: replaced 0 by nullptr Signed-off-by: Kenzo Lobos-Tsunekawa <[email protected]> * chore: removed unused headers Signed-off-by: Kenzo Lobos-Tsunekawa <[email protected]> * chore: removed double slash Signed-off-by: Kenzo Lobos-Tsunekawa <[email protected]> * chore: removed unused headers Signed-off-by: Kenzo Lobos-Tsunekawa <[email protected]> * chore: ci/cd Signed-off-by: Kenzo Lobos-Tsunekawa <[email protected]> * chore: ci/cd Signed-off-by: Kenzo Lobos-Tsunekawa <[email protected]> * fix: forgot the noescept in the definition Signed-off-by: Kenzo Lobos-Tsunekawa <[email protected]> * chore: ci/cd Signed-off-by: Kenzo Lobos-Tsunekawa <[email protected]> --------- Signed-off-by: Kenzo Lobos-Tsunekawa <[email protected]>
Description
Implemented a tensorrt plugin for the
argsort
operation.ONNX and TensorRT supports this operation via the TopK operation, but TensorRT does not support large inputs, so I am directly implementing this operation as argsort
This PR is needed for #10600
Related links
Parent Issue:
How was this PR tested?
Notes for reviewers
None.
Interface changes
None.
Effects on system behavior
None.