-
Notifications
You must be signed in to change notification settings - Fork 206
【CUDA Kernel No.39】collect_fpn_proposals算子Kernel修复 #2096
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
Conversation
|
Thanks for your contribution! |
YqGe585
left a comment
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.
可以考虑给iluvatar注册上这个kernel。
22fdb8a to
2d9e8b3
Compare
|
|
||
| #include "paddle/phi/kernels/gpu/collect_fpn_proposals_kernel.h" | ||
|
|
||
| PD_CUSTOM_KERNEL_REGISTER(collect_fpn_proposals, |
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.
这个kernel不要注册,因为天数不支持此算子
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.
好的,已经删除
74a8de4 to
74334e8
Compare
YqGe585
left a comment
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.
LGTM
backends\metax_gpu\kernels\cuda_kernels\collect_fpn_proposals_kernel_register.cubackends\metax_gpu\CMakeLists.txt列表中已有${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/collect_fpn_proposals_kernel.cu,无需新增backends\iluvatar_gpu\CMakeLists.txt列表中有多处${PADDLE_SOURCE_DIR}/paddle/phi/kernels/gpu/collect_fpn_proposals_kernel.cu