ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-06-29 11:16:59 +00:00
Code Issues Packages Projects Releases Wiki Activity

clang-format

Browse Source
This commit is contained in:
danyao12
2024-09-19 11:28:47 +00:00
parent 5ab137f44c
commit c3b406d6e8
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -93,7 +93,9 @@ auto create_args(int argc, char* argv[])
"if set to 1 will use multi-buffer reduction strategy for dq, atomic opeartion "
"will not be used")
.insert("ext_asm", "0", "if set to 1, some cases will call the ext asm dqdkdv kernel")
.insert("asm_atomic_fp32", "1", "if set to 0, atomic fp16/bf16 is used when calling the ext asm dqdkdv kernel");
.insert("asm_atomic_fp32",
"1",
"if set to 0, atomic fp16/bf16 is used when calling the ext asm dqdkdv kernel");
bool result = arg_parser.parse(argc, argv);
return std::make_tuple(result, arg_parser);