-
Notifications
You must be signed in to change notification settings - Fork 3.4k
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
[TOPI] Add GPU SSD #1397
[TOPI] Add GPU SSD #1397
Conversation
Supporting vision operators for gpu is great. But it is less flexible with another copy of operators written with ir_builder. Is it possible to create schedules for IR? Or as suggested in #1322, vision operators should be migrated to use hybrid script? If these are not possible, a rewrite of the IR is fine. |
We will migrate ssd related ops to hydrid frontend. IR builder version serves for the first stable version to support ssd on GPU. |
@were @srkreddy1238 please review |
Thanks @Laurawly for the contribution, this is now merged |
This PR adds the following ops:
nms including GPU sorting
multibox_prior for GPUs
and multibox_detection
Tests for individual ops can be found in topi/tests/python/test_topi_vision.py
@tqchen @kevinthesun @yzhliu @yidawang plz help review or comment, thanks!