Add fp8<->fp16 to PassThrough element op

2e7e564a · Rostyslav Geyyer · cf0845ac · 2e7e564a
Commit 2e7e564a authored May 24, 2023 by Rostyslav Geyyer
Hide whitespace changes
Inline Side-by-side

Showing with 12 additions and 0 deletions

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

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
@@ -99,6 +99,18 @@ struct PassThrough
    {
        y = type_convert<f8_t>(x);
    }
+
+    template <>
+    __host__ __device__ void operator()<half_t, f8_t>(half_t& y, const f8_t& x) const
+    {
+        y = type_convert<half_t>(x);
+    }
+
+    template <>
+    __host__ __device__ void operator()<f8_t, half_t>(f8_t& y, const half_t& x) const
+    {
+        y = type_convert<f8_t>(x);
+    }
 };

 struct UnaryConvert