ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-06-07 00:04:37 +00:00
Code Issues Packages Projects Releases Wiki Activity

Use 128 as minimus page_block_size

Browse Source
This commit is contained in:
PoYen, Chen
2024-08-06 03:20:29 +00:00
parent f9e2bafd10
commit 12da00c3be
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
example/ck_tile/01_fmha/fmha_fwd.cpp
View File

@@ -441,9 +441,9 @@ bool run(const ck_tile::ArgParser& arg_parser)
page_block_size = 0;
}
#endif
if(!(page_block_size % 256 == 0))
if(!(page_block_size % 128 == 0))
{
std::cerr << "only paged-kvcache block size divisible by 256 are currently supported"
std::cerr << "only paged-kvcache block size divisible by 128 are currently supported"
<< std::endl;
return false;
}