From 4ae9acd712258f8532d0342f8a2832825c92b576 Mon Sep 17 00:00:00 2001
From: Qianfeng Zhang <Qianfeng.Zhang@amd.com>
Date: Fri, 25 Apr 2025 14:42:33 +0000
Subject: [PATCH] Revert "Update exp() in ck_tile/core/numeric/math.hpp to use
 __expf"

This reverts commit 7316a44ff3310ddb331cb1af63c4d5f7ac0294db.
---
 include/ck_tile/core/numeric/math.hpp | 6 ++----
 1 file changed, 2 insertions(+), 4 deletions(-)

diff --git a/include/ck_tile/core/numeric/math.hpp b/include/ck_tile/core/numeric/math.hpp
index 56931fe795..6bdcb509b0 100644
--- a/include/ck_tile/core/numeric/math.hpp
+++ b/include/ck_tile/core/numeric/math.hpp
@@ -1359,8 +1359,7 @@ CK_TILE_DEVICE T rcp(T x)
 template <typename T>
 CK_TILE_DEVICE T exp(T x)
 {
-    // __expf in llvm using __builtin_amdgcn_exp2f, which is very fast
-    return type_convert<T>(__expf(type_convert<float>(x)));
+    return type_convert<T>(__ocml_exp_f32(type_convert<float>(x)));
 };
 
 template <>
@@ -1372,8 +1371,7 @@ CK_TILE_DEVICE fp16_t exp<fp16_t>(fp16_t x)
 template <>
 CK_TILE_DEVICE float exp<float>(float x)
 {
-    // __expf in llvm using __builtin_amdgcn_exp2f, which is very fast
-    return __expf(x);
+    return __ocml_exp_f32(x);
 };
 
 template <>