Commit cf2490e0 authored by fsx950223's avatar fsx950223
Browse files

fix name

parent db5310b0
...@@ -965,7 +965,7 @@ inline __host__ __device__ constexpr float type_convert<float, bhalf_t>(bhalf_t ...@@ -965,7 +965,7 @@ inline __host__ __device__ constexpr float type_convert<float, bhalf_t>(bhalf_t
} }
// convert fp32 to bfp16 // convert fp32 to bfp16
struct uint16_t_2{ struct uint16x2_t{
ushort x; ushort x;
ushort y; ushort y;
}; };
...@@ -976,7 +976,7 @@ inline __host__ __device__ constexpr bhalf_t type_convert<bhalf_t, float>(float ...@@ -976,7 +976,7 @@ inline __host__ __device__ constexpr bhalf_t type_convert<bhalf_t, float>(float
union union
{ {
float fp32; float fp32;
uint16_t_2 int32; uint16x2_t int32;
} u = {x}; } u = {x};
return u.int32.y + ((u.int32.x & 0x8000) >> 15); return u.int32.y + ((u.int32.x & 0x8000) >> 15);
......
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