fa4-cute-runtime / README.md
liangsu9988's picture
Update README.md
6ad28b7 verified
|
Raw
History Blame
728 Bytes
# fa4-cute-runtime
Forward-only, SM100-family FlashAttention-4 CuTe DSL package used by FlashRT's
production GROOT N1.7 and PI0.5 Thor runtimes. In addition to the standard
D48/D72/D128 path, it includes the dedicated D256 2CTA forward kernel needed
by PI0.5's PaliGemma encoder.
The package exposes `flash_attn_func`, `flash_attn_varlen_func`, and the
preallocated `forward_static` CUDA Graph entry, including `seqused_k` for
fixed-shape padded K/V. It keeps the vendored
implementation in the private `flashrt_fa4` namespace and vendors the required
quack helper subset, so it cannot replace or mutate upstream `flash_attn`
imports.
See [CARD.md](CARD.md) for usage and [VALIDATION.md](VALIDATION.md) for the
release gate.