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

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

Browse Source
This commit is contained in:
Qianfeng Zhang
2025-04-21 14:39:42 +00:00
parent 26db7e0b7c
commit 7316a44ff3
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -1359,7 +1359,8 @@ CK_TILE_DEVICE T rcp(T x)
template <typename T>
CK_TILE_DEVICE T exp(T x)
{
return type_convert<T>(__ocml_exp_f32(type_convert<float>(x)));
// __expf in llvm using __builtin_amdgcn_exp2f, which is very fast
return type_convert<T>(__expf(type_convert<float>(x)));
};
template <>
@@ -1371,7 +1372,8 @@ CK_TILE_DEVICE fp16_t exp<fp16_t>(fp16_t x)
template <>
CK_TILE_DEVICE float exp<float>(float x)
{
return __ocml_exp_f32(x);
// __expf in llvm using __builtin_amdgcn_exp2f, which is very fast
return __expf(x);
};
template <>