GitHub-Proxy/eigen
1
0
Fork 0
mirror of https://gitlab.com/libeigen/eigen.git synced 2025-04-30 15:54:13 +08:00
Code Issues Packages Projects Releases Wiki Activity

Implemented some of the missing type casting for half floats

Browse Source
This commit is contained in:
Benoit Steiner 2016-03-17 21:45:45 -07:00
parent 95b8961a9b
commit 7b98de1f15
1 changed files with 3 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
Eigen/src/Core/arch/CUDA/TypeCasting.h
View File

@ -24,8 +24,7 @@ struct scalar_cast_op<float, half> {
#if defined(__CUDA_ARCH__) && __CUDA_ARCH__ >= 300 #if defined(__CUDA_ARCH__) && __CUDA_ARCH__ >= 300
return __float2half(a); return __float2half(a);
#else #else
assert(false && "tbd"); return half(a);
return half();
#endif #endif
} }
}; };
@ -43,8 +42,7 @@ struct scalar_cast_op<int, half> {
#if defined(__CUDA_ARCH__) && __CUDA_ARCH__ >= 300 #if defined(__CUDA_ARCH__) && __CUDA_ARCH__ >= 300
return __float2half(static_cast<float>(a)); return __float2half(static_cast<float>(a));
#else #else
assert(false && "tbd"); return half(static_cast<float>(a));
return half();
#endif #endif
} }
}; };
@ -62,8 +60,7 @@ struct scalar_cast_op<half, float> {
#if defined(__CUDA_ARCH__) && __CUDA_ARCH__ >= 300 #if defined(__CUDA_ARCH__) && __CUDA_ARCH__ >= 300
return __half2float(a); return __half2float(a);
#else #else
assert(false && "tbd"); return static_cast<float>(a);
return 0.0f;
#endif #endif
} }
}; };