Update UnaryConvertPrecision

0a622886 · Rosty Geyyer · 25b7bd72 · 0a622886
Commit 0a622886 authored Apr 19, 2023 by Rosty Geyyer
Hide whitespace changes
Inline Side-by-side

Showing with 5 additions and 2 deletions

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

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
@@ -86,10 +86,13 @@ struct UnaryConvert
    }
 };

-struct UnaryConvertPrecision : UnaryConvert
+struct UnaryConvertPrecision
 {
    template <typename Y, typename X>
-    __host__ __device__ void operator()(Y& y, const X& x) const;
+    __host__ __device__ void operator()(Y& y, const X& x) const
+    {
+        y = type_convert<Y>(x);
+    }

    template <>
    __host__ __device__ void operator()<bhalf_t, float>(bhalf_t& y, const float& x) const