Uploaded using `kernel-builder`.
Browse files
README.md
CHANGED
|
@@ -12,27 +12,15 @@ This is the repository card of kernels-community/deformable-detr that has been p
|
|
| 12 |
from kernels import get_kernel
|
| 13 |
|
| 14 |
kernel_module = get_kernel("kernels-community/deformable-detr")
|
| 15 |
-
|
| 16 |
|
| 17 |
-
|
| 18 |
```
|
| 19 |
|
| 20 |
## Available functions
|
| 21 |
-
- `layers`
|
| 22 |
- `ms_deform_attn_forward`
|
| 23 |
- `ms_deform_attn_backward`
|
| 24 |
|
| 25 |
## Benchmarks
|
| 26 |
|
| 27 |
Benchmarking script is available for this kernel. Run `kernels benchmark kernels-community/deformable-detr`.
|
| 28 |
-
|
| 29 |
-
### Performance
|
| 30 |
-
|
| 31 |
-
<img class="dark:hidden border border-gray-200 dark:border-gray-700 rounded-lg" src="media/benches_light_animation.svg" />
|
| 32 |
-
<img class="hidden dark:block border border-gray-200 dark:border-gray-700 rounded-lg" src="media/benches_dark_animation.svg" />
|
| 33 |
-
|
| 34 |
-
<img class="dark:hidden border border-gray-200 dark:border-gray-700 rounded-lg" src="media/benches_light_latency.svg" />
|
| 35 |
-
<img class="hidden dark:block border border-gray-200 dark:border-gray-700 rounded-lg" src="media/benches_dark_latency.svg" />
|
| 36 |
-
|
| 37 |
-
<img class="dark:hidden border border-gray-200 dark:border-gray-700 rounded-lg" src="media/benches_light_throughput.svg" />
|
| 38 |
-
<img class="hidden dark:block border border-gray-200 dark:border-gray-700 rounded-lg" src="media/benches_dark_throughput.svg" />
|
|
|
|
| 12 |
from kernels import get_kernel
|
| 13 |
|
| 14 |
kernel_module = get_kernel("kernels-community/deformable-detr")
|
| 15 |
+
ms_deform_attn_forward = kernel_module.ms_deform_attn_forward
|
| 16 |
|
| 17 |
+
ms_deform_attn_forward(...)
|
| 18 |
```
|
| 19 |
|
| 20 |
## Available functions
|
|
|
|
| 21 |
- `ms_deform_attn_forward`
|
| 22 |
- `ms_deform_attn_backward`
|
| 23 |
|
| 24 |
## Benchmarks
|
| 25 |
|
| 26 |
Benchmarking script is available for this kernel. Run `kernels benchmark kernels-community/deformable-detr`.
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|