Skip to content

GitLab

Commit 06e287e8 authored by letaoqin's avatar letaoqin
Browse files

fix precision

parent 1e447622
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
include/ck/tensor_operation/gpu/element/binary_element_wise_operation.hpp
View file @ 06e287e8
......@@ -47,14 +47,14 @@ struct Add
__host__ __device__ constexpr void
operator()<half_t>(half_t& y, const float& x0, const half_t& x1) const
{
y = type_convert<half_t>(x0) + x1;
y = type_convert<half_t>(type_convert<float>(x1) + x0);
};
template <>
__host__ __device__ constexpr void
operator()<half_t>(half_t& y, const half_t& x0, const half_t& x1) const
{
y = x0 + x1;
y = type_convert<float>(x0) + type_convert<float>(x1);
};
template <>
......
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