| --- |
| library_name: kernels |
| license: apache-2.0 |
| --- |
| |
| <!-- This model card has been generated automatically according to the information the training script had access to. You |
| should probably proofread and complete it, then remove this comment. --> |
|
|
|
|
| This is the repository card of {repo_id} that has been pushed on the Hub. It was built to be used with the [`kernels` library](https://github.com/huggingface/kernels). This card was automatically generated. |
| |
| |
| ## How to use |
| |
| ```python |
| # make sure `kernels` is installed: `pip install -U kernels` |
| from kernels import get_kernel |
|
|
| kernel_module = get_kernel("REPO_ID") # <- change the ID if needed |
| flash_attn_combine = kernel_module.flash_attn_combine |
|
|
| flash_attn_combine(...) |
| ``` |
| |
| ## Available functions |
| |
| - `flash_attn_combine` |
| - `flash_attn_func` |
| - `flash_attn_qkvpacked_func` |
| - `flash_attn_varlen_func` |
| - `flash_attn_with_kvcache` |
| - `get_scheduler_metadata` |
| |
| ## Supported backends |
| |
| - cuda |
| |
| ## CUDA Capabilities |
| |
| - 9.0a |
| - 8.0 |
| |
| ## Benchmarks |
| |
| [TODO: provide benchmarks if available] |
| |
| ## Code source |
| |
| [TODO: provide original code source and other relevant citations if available] |
| |
| ## Notes |
| |
| [TODO: provide additional notes about this kernel if needed] |
