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

Enlarge rotary_dim limit (8 -> 16)

Browse Source
This commit is contained in:
PoYen, Chen
2024-07-26 06:51:24 +00:00
parent 4280a07d2a
commit d41ff70db5
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

@@ -409,9 +409,9 @@ bool run(const ck_tile::ArgParser& arg_parser)
std::cerr << "rotary_dim should be less than or equal to head dim for q" << std::endl;
return false;
}
else if(!(rotary_dim % 8 == 0))
else if(!(rotary_dim % 16 == 0))
{
std::cerr << "only rotary dimensions divisible by 8 are currently supported" << std::endl;
std::cerr << "only rotary dimensions divisible by 16 are currently supported" << std::endl;
return false;
}