| | --- |
| | 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] |
