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

[CK_TILE] FMHA BWD launcher: address PR #7331 review comments

Browse Source
This commit is contained in:
Ding, Yi
2026-05-12 02:46:41 -05:00
parent d434410e52
commit 1f4cc34e68
3 changed files with 27 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -323,7 +323,7 @@ struct FmhaBwdDQDKDVKernel
std::forward<Args>(args)...);
}
template <typename... Args>
CK_TILE_HOST static constexpr auto GetWorkspaceDeviceSizeUpperBound(Args&&... args)
CK_TILE_HOST static size_t GetWorkspaceDeviceSizeUpperBound(Args&&... args)
{
return WorkspaceManager::template GetWorkspaceDeviceSizeUpperBound<
kUseQrQtrDorPipeline,