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

Fix type errors in composes<>

Browse Source
This commit is contained in:
Po Yen Chen
2024-04-09 13:18:17 +00:00
parent 35e2d18e5e
commit c6eac9746f
2 changed files with 8 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -508,7 +508,7 @@ bool run(const ck_tile::ArgParser& arg_parser)
s_host_ref,
ck_tile::identity{},
ck_tile::identity{},
[&](SaccDataType x) { return pcompute_element_func(scale * x); });
ck_tile::composes(pcompute_element_func, ck_tile::scales(scale)));
if(use_bias)
{