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

Use type_convert rather than static_cast in f_silu

Browse Source
This commit is contained in:
Qianfeng Zhang
2025-05-07 07:05:43 +00:00
parent 72d55d1b40
commit 079f7e3a03
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/18_hstu_attention/hstu_attention_fwd_pipeline.hpp
View File

@@ -239,11 +239,11 @@ struct HstuAttentionFwdPipelineQRKSVS
if constexpr(std::is_same_v<CompDataType, float>)
{
x = x * __builtin_amdgcn_rcpf(neg_one - __expf(x)) *
__builtin_amdgcn_rcpf(static_cast<CompDataType>(max_seqlen));
__builtin_amdgcn_rcpf(type_convert<CompDataType>(max_seqlen));
}
else
{
x = x / (neg_one - exp(x)) / static_cast<CompDataType>(max_seqlen);
x = x / (neg_one - exp(x)) / type_convert<CompDataType>(max_seqlen);
}
};