fix F16

22e7a408 · ltqin · f1ed4c5e · 22e7a408
Commit 22e7a408 authored Mar 08, 2023 by ltqin
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

include/ck/tensor_operation/gpu/element/element_wise_operation.hpp ...k/tensor_operation/gpu/element/element_wise_operation.hpp +1 -1

No files found.
--- a/include/ck/tensor_operation/gpu/element/element_wise_operation.hpp
+++ b/include/ck/tensor_operation/gpu/element/element_wise_operation.hpp
@@ -423,7 +423,7 @@ struct ScaleBiasMask

    template <>
    __host__ __device__ constexpr void
-    operator()(float& y, const float& x, const F16& bias, const int32_t& mask) const
+    operator()(float& y, const float& x, const half_t& bias, const int32_t& mask) const
    {
        float filter_value = (mask == 1 ? 0.0f : mask_filter_value_);
        y                  = scale_ * x + ck::type_convert<float>(bias) + filter_value;