attn-decode / README.md

Add kernel card (library_name: kernels)

2d2dfa8 verified 9 days ago

604 Bytes

library_name: kernels
tags:
  - kernels
  - rocm

Flash-Attention Decode (RDNA4/gfx1201)

Dense, paged, and fp8 flash-attention decode kernels (GQA, sliding-window, block-table KV cache).

Built with kernel-builder for AMD RDNA4 (gfx1201). Load with the kernels library:

from kernels import get_kernel
kernel = get_kernel("pat883/attn-decode")

Requires a ROCm PyTorch build (torch 2.10 / ROCm 7.x) on an RDNA4 card. Built variants: torch210-cxx11-rocm70 and torch210-cxx11-rocm71.