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