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

Merge commit '92653168c2b276d4467320f5bdff5ec6cbddf4e6' into develop

Browse Source
This commit is contained in:
assistant-librarian[bot]
2025-12-17 17:16:17 +00:00
parent 28b493d331
commit c5faecd894
11 changed files with 309 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -12,6 +12,8 @@ static CK_TILE_HOST_DEVICE constexpr index_t ceil_to_qualified_tile_length()
{
if constexpr(Headdim == 48)
return 48;
else if constexpr(Headdim == 80)
return 96;
else if constexpr(Headdim == 96)
return 128;
else if constexpr(Headdim == 160)