search

PaddlePaddle / Paddle3D

A 3D computer vision development toolkit based on PaddlePaddle. It supports point-cloud object detection, segmentation, and monocular 3D object detection models.
Apache License 2.0
573 stars 141 forks source link

Optimize ms deform attention #416

Closed Vvsmile closed 8 months ago

Vvsmile commented 1 year ago

Optimize the ms_deform_attention.

Mainly optimize this operator by replacing serial reduction by warp reduction. Passingly, some computation like divided or mod by multiply of 2, the redundancy of computing const value are optimized meanwhile.

paddle-bot[bot] commented 1 year ago

Thanks for your contribution!

nepeplwu commented 9 months ago

This operator implementation seems unable to fully align the forward and backward outputs of the original implementation.