Use ck math version of exp

af2004b5 · rocking · 0d02a5c7 · af2004b5
Commit af2004b5 authored Apr 07, 2023 by rocking
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

include/ck/tensor_operation/gpu/element/unary_element_wise_operation.hpp ...or_operation/gpu/element/unary_element_wise_operation.hpp +1 -1

No files found.
--- a/include/ck/tensor_operation/gpu/element/unary_element_wise_operation.hpp
+++ b/include/ck/tensor_operation/gpu/element/unary_element_wise_operation.hpp
@@ -342,7 +342,7 @@ struct Swish
                          is_same<T, ck::half_t>::value,
                      "Data type is not supported by this operation!");
-        y = x / (ck::type_convert<T>(1) + exp(-beta_ * x));
+        y = x / (ck::type_convert<T>(1) + ck::math::exp(-beta_ * x));
    };
    float beta_ = 1.0f;