Skip to content

LW-DETR multi-scale deformable attention CUDA op - #33

Merged
cregouby merged 6 commits into
mlverse:mainfrom
srishtiii28:ms-deform-attn
Aug 1, 2026
Merged

LW-DETR multi-scale deformable attention CUDA op#33
cregouby merged 6 commits into
mlverse:mainfrom
srishtiii28:ms-deform-attn

Conversation

@srishtiii28

@srishtiii28 srishtiii28 commented Jul 7, 2026

Copy link
Copy Markdown
Contributor

Adds ops_ms_deform_attn(), the multi-scale deformable attention CUDA op LW-DETR needs (fix #25). Replaces the slow pure-R grid_sample fallback the decoder's cross-attention uses on GPU.

@cregouby
cregouby self-requested a review July 9, 2026 16:32
cregouby and others added 3 commits July 11, 2026 16:40
…erse#35)

* bump zlib, libpng, libjpeg to package compilation

* add fs libuv1 dependancy to CI-CD

* bump checkout,  Jimver/cuda-toolkit,  setup-pandoc , download-artifact,   upload-artifact
@cregouby
cregouby merged commit 2e829b5 into mlverse:main Aug 1, 2026
6 of 7 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

implement multi-scale deformable attention CUDA ops

2 participants