Commit 5aff1a05 authored by fsx950223's avatar fsx950223
Browse files

move uint16x2_t to function body

parent a61aa740
......@@ -965,14 +965,14 @@ inline __host__ __device__ constexpr float type_convert<float, bhalf_t>(bhalf_t
}
// convert fp32 to bfp16
struct uint16x2_t{
ushort x;
ushort y;
};
template <>
inline __host__ __device__ constexpr bhalf_t type_convert<bhalf_t, float>(float x)
{
struct uint16x2_t{
ushort x;
ushort y;
};
union
{
float fp32;
......
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