CK mha bwd: add sink attention score gradient support - #2321

Merged

poyenc merged 16 commits into

mainfrom

lj_ck_sink_bwd_v2

Apr 4, 2026

Commits on Mar 18, 2026

CK mha bwd: add sink attention score gradient support
LJ-underdog
committed
d3fd124
test: add varlen sink bwd tests to test_mha_sink_bwd
LJ-underdog
committed
d974b3e
Update op_tests/test_mha_sink_bwd.py

LJ-underdog
and
github-actions[bot]
authored
b065c60
style: apply black formatting to test_mha_sink_bwd
LJ-underdog
committed
a7bf4ae
test: move sink bwd tests into test_mha.py and test_mha_varlen.py
LJ-underdog
committed
4ac64db
style: apply black formatting to sink bwd tests in test_mha and test_mha_varlen
LJ-underdog
committed
3690ad1

Commits on Apr 1, 2026

Merge branch 'main' into lj_ck_sink_bwd_v2
LJ-underdog
authored
b559c5a

Commits on Apr 2, 2026

fix: adapt mha bwd to updated CK fmha_bwd API and zero dq_accum
LJ-underdog
committed
35200bf
update ck to ROCm/rocm-libraries#5504
LJ-underdog
committed
7481fd6
Revert "update ck to ROCm/rocm-libraries#5504"
LJ-underdog
committed
852a009
update ck commit
LJ-underdog
committed
aaf8519
Merge remote-tracking branch 'origin/main' into lj_ck_sink_bwd_v2
LJ-underdog
committed
81baf75
update bwd args
LJ-underdog
committed
d93c1b2

Commits on Apr 3, 2026