SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Use deformable_detr kernel from the Hub (#36853)

Browse Source 
* Use `deformable_detr` kernel from the Hub

Remove the `deformable_detr` kernel from `kernels/` and use the
pre-built kernel from the Hub instead.

* Add license header

* Add `kernels` as an extra `hub-kernels`

Also add it to `testing`, so that the kernel replacement gets tested
when using CUDA in CI.
This commit is contained in:
Daniël de Kok
2025-03-21 13:08:47 +01:00
committed by GitHub
parent 2638d54e78
commit f94b0c59f2
21 changed files with 409 additions and 3838 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
utils/check_build.py
View File

@@ -21,8 +21,6 @@ from pathlib import Path
FILES_TO_FIND = [
"kernels/rwkv/wkv_cuda.cu",
"kernels/rwkv/wkv_op.cpp",
"kernels/deformable_detr/ms_deform_attn.h",
"kernels/deformable_detr/cuda/ms_deform_im2col_cuda.cuh",
"kernels/falcon_mamba/selective_scan_with_ln_interface.py",
"kernels/falcon_mamba/__init__.py",
"kernels/__init__.py",