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

Revert "Update exp() in ck_tile/core/numeric/math.hpp to use __expf"

Browse Source 
This reverts commit 7316a44ff3.
This commit is contained in:
Qianfeng Zhang
2025-04-25 14:42:33 +00:00
parent 80677eb6e0
commit 4ae9acd712
1 changed files with 2 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
include/ck_tile/core/numeric/math.hpp
View File

@@ -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 <>