ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-13 17:55:48 +00:00
Code Issues Packages Projects Releases Wiki Activity

[CK_TILE] Use Unified Workspace for FMHA BWD

Browse Source
This commit is contained in:
Ding, Yi
2026-04-03 02:10:43 -05:00
parent 2510e7b238
commit 28afc8fee3
5 changed files with 527 additions and 324 deletions
Download Patch File Download Diff File Expand all files Collapse all files

631
include/ck_tile/ops/fmha/kernel/fmha_bwd_kernel.hpp
View File

File diff suppressed because it is too large Load Diff

2
include/ck_tile/ops/fmha/pipeline/block_fmha_bwd_pipeline_problem.hpp
View File

@@ -95,7 +95,6 @@ template <typename AccDataType_,
typename QGradDataType_,
index_t kBlockSize_,
index_t kM0_,
index_t kN0_,
index_t kQKHeaddim_,
bool kIsGroupMode_,
bool kIsDeterministic_,
@@ -111,7 +110,6 @@ struct BlockFmhaBwdConvertQGradPipelineProblem
static constexpr index_t kBlockSize = kBlockSize_;
static constexpr index_t kM0 = kM0_;
static constexpr index_t kN0 = kN0_;
static constexpr index_t kQKHeaddim = kQKHeaddim_;
static constexpr bool kIsGroupMode = kIsGroupMode_;
static constexpr bool kIsDeterministic = kIsDeterministic_;