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

Set num_splits=1 if split-kv is not supported

Browse Source
This commit is contained in:
PoYen, Chen
2024-07-08 10:27:32 +00:00
parent 6ca3910199
commit 18a3834fb4
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -370,6 +370,7 @@ bool run(const ck_tile::ArgParser& arg_parser)
if(num_splits != 1)
{
std::cerr << "split-kv is not supported. ignoring the 'num_splits' option" << std::endl;
num_splits = 1;
}
#endif