devtools/eigen
1
0
Fork 0
mirror of https://gitlab.com/libeigen/eigen.git synced 2026-04-10 11:34:33 +08:00
Code Issues Packages Projects Releases Wiki Activity

Explicitly specify the rounding mode when converting floats to fp16

Browse Source
This commit is contained in:
Benoit Steiner
2016-05-25 15:56:15 -07:00
parent 3ac4045272
commit 0322c66a3f
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
Eigen/src/Core/arch/CUDA/PacketMathHalf.h
View File

@@ -378,7 +378,7 @@ ploadquad<Packet8h>(const Eigen::half* from) {
EIGEN_STRONG_INLINE Packet8f half2float(const Packet8h& a) { EIGEN_STRONG_INLINE Packet8f half2float(const Packet8h& a) {
#ifdef EIGEN_HAS_FP16_C #ifdef EIGEN_HAS_FP16_C
return _mm256_cvtph_ps(a.x); return _mm256_cvtph_ps(a.x, _MM_FROUND_TO_NEAREST_INT|_MM_FROUND_NO_EXC);
#else #else
EIGEN_ALIGN32 Eigen::half aux[8]; EIGEN_ALIGN32 Eigen::half aux[8];
pstore(aux, a); pstore(aux, a);