Downloads · 30 days
4
12% of all-time downloads
attn-hip
pat883/attn-hip
attn-hip is a machine learning model from pat883. Use it for the machine learning task on the model card, and read the license before you ship it in a product. It is set up for kernels.
Dense flash-attention prefill (causal + sliding-window + additive mask) as a native HIP + rocWMMA kernel.
All-time downloads
34
Public
Repo size
11.9 MB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.so4 MB · 100%
At a glance
- Library
- kernels
- Access
- Public
- Created
- Jul 3, 2026
- Updated
- Jul 3, 2026
- SHA
- d8152a8d
Model card
From the Hugging Face model README
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:
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-rocm70andtorch210-cxx11-rocm71.
- Library
- kernels
- Created
- Jul 3, 2026
- Updated
- Jul 3, 2026