Kernels:

---

amd

/

aiter-sage-attention

Like 0

Follow

AMD 3.44k

Trusted publisher

This is the repository card of kernels-community/aiter-sage-attention that has been pushed on the Hub. It was built to be used with the kernels library. This card was automatically generated.

How to use

# make sure `kernels` is installed: `pip install -U kernels`
from kernels import get_kernel

kernel_module = get_kernel("kernels-community/aiter-sage-attention", version=1)
gfx1201_sage_attention = kernel_module.gfx1201_sage_attention

gfx1201_sage_attention(...)

Available functions

• gfx1201_sage_attention
• gfx1201_sage_attention_fwd
• gfx1201_sage_prepare
• sageattn

Benchmarks

No benchmark available yet.