diff --git a/example/ck_tile/18_hstu_attention/hstu_attention_fwd_pipeline.hpp b/example/ck_tile/18_hstu_attention/hstu_attention_fwd_pipeline.hpp index ea8d60ef6f..a0ce38c92d 100644 --- a/example/ck_tile/18_hstu_attention/hstu_attention_fwd_pipeline.hpp +++ b/example/ck_tile/18_hstu_attention/hstu_attention_fwd_pipeline.hpp @@ -247,7 +247,7 @@ struct HstuAttentionFwdPipelineQRKSVS const auto f_silu = [](CompDataType& x) { const auto neg_one = ck_tile::type_convert(-1.0f); - return x = x / (neg_one - exp(x)); + x = x * __builtin_amdgcn_rcpf(neg_one - exp(x)); }; using OaccBlockTileType = decltype(gemm_1.MakeCBlockTile());