ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-27 08:25:46 +00:00
Code Issues Packages Projects Releases Wiki Activity

[CK_TILE] FMHA BWD Enable Tile 16x192 (#2741)

Browse Source 
* 16x192

* Use buffer_load_lds for lse/d

* Dispatch & cleanup

* Avoid zeroing dq & fix

* fix

[ROCm/composable_kernel commit: ead4447b20]
This commit is contained in:
Yi DING
2025-08-28 18:54:18 +08:00
committed by GitHub
parent 311fd3d194
commit f2e6edde3b
6 changed files with 173 additions and 114 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
example/ck_tile/01_fmha/fmha_bwd.cpp
View File

@@ -803,7 +803,6 @@ bool run(const ck_tile::ArgParser& arg_parser)
o_buf.ToDevice(o_host.data());
lse_buf.ToDevice(lse_host.data());
dq_buf.SetZero();
dbias_buf.SetZero();
dq_acc_buf.SetZero();