Skip to content

GitLab

"include/vscode:/vscode.git/clone" did not exist on "fb6dafee1284df6af6590d47a8590f6bf4ab6b0c"
Commit 0a622886 authored by Rosty Geyyer's avatar Rosty Geyyer
Browse files

Update UnaryConvertPrecision

parent 25b7bd72
Show whitespace changes
Inline Side-by-side
Showing with 5 additions and 2 deletions
include/ck/tensor_operation/gpu/element/unary_element_wise_operation.hpp
View file @ 0a622886
...@@ -86,10 +86,13 @@ struct UnaryConvert ...@@ -86,10 +86,13 @@ struct UnaryConvert
} }
}; };
struct UnaryConvertPrecision : UnaryConvert struct UnaryConvertPrecision
{ {
template <typename Y, typename X> 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 <> template <>
__host__ __device__ void operator()<bhalf_t, float>(bhalf_t& y, const float& x) const __host__ __device__ void operator()<bhalf_t, float>(bhalf_t& y, const float& x) const
......
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