Repository navigation
[docs] kernels - #13139
[docs] kernels#13139
Conversation
sayakpaul
left a comment
There was a problem hiding this comment.
Thanks a lot for prioritizing it.
|
|
||
| [Kernels](https://huggingface.co/docs/kernels/index) is a library for building, distributing, and loading optimized compute kernels on the [Hub](https://huggingface.co/kernels-community). It supports [attention](./attention_backends#set_attention_backend) kernels and custom CUDA kernels for operations like RMSNorm. | ||
|
|
||
| The [Diffusers Pipeline Integration](https://github.com/huggingface/kernels/blob/main/skills/cuda-kernels/references/diffusers-integration.md) guide shows how to integrate a kernel. Create a custom optimized attention processor, patch all modules in the model, and inject the kernel into the pipeline. |
There was a problem hiding this comment.
The kernel skill basically lets users get an agent to write custom kernels for a model and hardware. It's not specific to the attention processor but also other modules as well such RMSNorm. Should we make it clearer?
There was a problem hiding this comment.
lmk if this is clearer!
There was a problem hiding this comment.
It wasn't just RMSNorm but also other modules implemented with custom kernels.
There was a problem hiding this comment.
i mention RMSNorm as an example only for the benchmark results below
There was a problem hiding this comment.
Thanks! I would like to also see what @burtenshaw thinks about this.
* kernels * feedback

adds a kernels section in the
Accelerate inferencedocs with the results:Attention backendsdocs which demonstrates support for loading attention kernels withset_attention_backend