From 022ed3fd8a211cf8846a29a10f2e89cc8f85f837 Mon Sep 17 00:00:00 2001
From: Qianfeng Zhang <Qianfeng.Zhang@amd.com>
Date: Mon, 21 Apr 2025 14:41:51 +0000
Subject: [PATCH] Back to use exp() instead of exp2() since exp() in ck_tile
 using fast __builtin_amdgcn_exp2f()

---
 .../ck_tile/18_hstu_attention/hstu_attention_fwd_pipeline.hpp   | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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 315379365e..cdc8fb6c3a 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 one = ck_tile::type_convert<CompDataType>(1.0f);
 
-            return x = x / (one + exp2(-log2e_v<CompDataType> * x));
+            return x = x / (one + exp(-x));
         };
 
         using OaccBlockTileType = decltype(gemm_1.MakeCBlockTile());