Fix: Correct/Improve the triton attention kernel#196

Open

chiennv2000 wants to merge 2 commits intoopenai:mainfrom

chiennv2000:attn_triton

Commits on Sep 23, 2025

fix: correct loop bounds for attention kernel
chiennv2000
committed

Commits on Sep 28, 2025

add: support backward pass, improve GQA by manipulating pointers instead of repeat_interleave
chiennv2000
committed