Skip to content

feat(kernels): SM90 TMA + mma.sync FlashAttention, forward + backward (causal, varlen, LSE)#237

Open
Billy1900 wants to merge 4 commits into
RL-Align:mainfrom
Billy1900:feat/sm90-attention-varlen-fwd
Open

feat(kernels): SM90 TMA + mma.sync FlashAttention, forward + backward (causal, varlen, LSE)#237
Billy1900 wants to merge 4 commits into
RL-Align:mainfrom
Billy1900:feat/sm90-attention-varlen-fwd

feat(kernels): SM90 backward (dQ/dK/dV) for varlen FlashAttention, fu…

013f4b4
Select commit
Loading
Failed to load commit list.
Sign in for the full log view
Job log options

This job was skipped