fix CvFp16Test failure

This commit is contained in:
Tomoaki Teshima 2018-09-25 15:00:37 +09:00
parent f4b9acb4db
commit 6a5266df79

View File

@ -288,7 +288,7 @@ template <> __device__ __forceinline__ short cast_fp16<float, short>(float v)
{ {
#if __CUDACC_VER_MAJOR__ >= 9 #if __CUDACC_VER_MAJOR__ >= 9
__half h(v); __half h(v);
return *(short*)&v; return *(short*)&h;
#else #else
return (short)__float2half_rn(v); return (short)__float2half_rn(v);
#endif #endif