为什么ms_deform_attn需要分别有一个CPU版本和GPU版本？

IDEA-Research / GroundingDINO

[ECCV 2024] Official implementation of the paper "Grounding DINO: Marrying DINO with Grounded Pre-Training for Open-Set Object Detection"

https://arxiv.org/abs/2303.05499

Apache License 2.0

6.83k stars 693 forks source link

Open YeHuiW opened 7 months ago

YeHuiW commented 7 months ago

为什么ms_deform_attn需要分别有一个CPU版本和GPU版本？我可以在GPU模式下使用 multi_scale_deformable_attn_pytorch 函数吗？这会带来性能下降吗？？