Skip to content

GitLab

Commit 22e7a408 authored by ltqin's avatar ltqin
Browse files

fix F16

parent f1ed4c5e
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
include/ck/tensor_operation/gpu/element/element_wise_operation.hpp
View file @ 22e7a408
...@@ -423,7 +423,7 @@ struct ScaleBiasMask ...@@ -423,7 +423,7 @@ struct ScaleBiasMask
template <> template <>
__host__ __device__ constexpr void __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_); float filter_value = (mask == 1 ? 0.0f : mask_filter_value_);
y = scale_ * x + ck::type_convert<float>(bias) + filter_value; y = scale_ * x + ck::type_convert<float>(bias) + filter_value;
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment