You can deploy this model and start earning money today!

pat883/
attn-hip

Name: attn-hip
Availability: InStock

Views

No views yet

Flash-Attention Prefill (RDNA4/gfx1201)

Dense flash-attention prefill (causal + sliding-window + additive mask) as a native HIP + rocWMMA kernel.

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

python

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

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