From aefd5fd5c4331f0265abb692d4742b558f13f01d Mon Sep 17 00:00:00 2001 From: nluehr Date: Tue, 28 Nov 2017 10:15:46 -0800 Subject: [PATCH] Replace __float2half_rn with __float2half The latter provides a consistent definition for CUDA 8.0 and 9.0. --- Eigen/src/Core/arch/CUDA/PacketMathHalf.h | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/Eigen/src/Core/arch/CUDA/PacketMathHalf.h b/Eigen/src/Core/arch/CUDA/PacketMathHalf.h index e3e45b03a..8a6f209c4 100644 --- a/Eigen/src/Core/arch/CUDA/PacketMathHalf.h +++ b/Eigen/src/Core/arch/CUDA/PacketMathHalf.h @@ -231,7 +231,7 @@ template<> __device__ EIGEN_STRONG_INLINE Eigen::half predux(const half2& #else float a1 = __low2float(a); float a2 = __high2float(a); - return Eigen::half(__float2half_rn(a1 + a2)); + return Eigen::half(__float2half(a1 + a2)); #endif } @@ -265,7 +265,7 @@ template<> __device__ EIGEN_STRONG_INLINE Eigen::half predux_mul(const ha #else float a1 = __low2float(a); float a2 = __high2float(a); - return Eigen::half(__float2half_rn(a1 * a2)); + return Eigen::half(__float2half(a1 * a2)); #endif }