ms_deform_attn_forward gpu support

Fangyi-Chen / SQR

MIT License

103 stars 5 forks source link

ms_deform_attn_forward gpu support #5

Closed Ahnsun closed 1 year ago

Ahnsun commented 1 year ago

Hi, I run the deformable-detr setting of SQR and get an error when applying ms_deform_attn_forward function. It seems like that the cuda op of deformable-detr is not deployed.