Kernels:
Trusted publisher
File size: 1,079 Bytes
c36a528 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 | ---
library_name: kernels
license: apache-2.0
---
<!-- This model card has automatically been generated. 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("kernels-community/sage-attention") # <- change the ID if needed
per_block_int8 = kernel_module.per_block_int8
per_block_int8(...)
```
## Available functions
- `per_block_int8`
- `per_warp_int8`
- `sub_mean`
- `per_channel_fp8`
- `sageattn`
## Supported backends
- cuda
## CUDA Capabilities
- 8.0
- 8.9
- 9.0a
## Benchmarks
[TODO: provide benchmarks if available]
## Source code
[TODO: provide original source code and other relevant citations if available]
## Notes
[TODO: provide additional notes about this kernel if needed] |