feat(hip): cascade attention support on ROCm using HIP - #221

Merged

demandal25 merged 6 commits into

AMD-Ecosystem:amd-integrationfrom

demandal25:cascade-attention

May 12, 2026

Commits on May 12, 2026

feat(hip): expose cascade attention on ROCm

demandal25
and
claude
committed
adcccd0
perf(hip): wave-64 launch geometry for cascade merge kernels on CDNA3

demandal25
and
claude
committed
dcf7fa9
feat(hip): fused cascade epilogue in BatchPrefillWithPagedKVCacheWrapper

demandal25
and
claude
committed
a18809c
test(hip): add cascade attention tests for ROCm

demandal25
and
claude
committed
de955a1
feat(rocm): cascade attention parity and test coverage

demandal25
and
claude
committed
e308e4c
fix(rocm): address Copilot review comments on cascade/prefill code

demandal25
and
claude
committed
dda0873