diff options
Diffstat (limited to 'Eigen/src/Core/arch/GPU/PacketMathHalf.h')
-rw-r--r-- | Eigen/src/Core/arch/GPU/PacketMathHalf.h | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/Eigen/src/Core/arch/GPU/PacketMathHalf.h b/Eigen/src/Core/arch/GPU/PacketMathHalf.h index 4b2c01022..5e143e65a 100644 --- a/Eigen/src/Core/arch/GPU/PacketMathHalf.h +++ b/Eigen/src/Core/arch/GPU/PacketMathHalf.h @@ -180,8 +180,10 @@ template <> EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE half2 pselect<half2>(const half2& mask, const half2& a, const half2& b) { - half result_low = __low2half(mask) == __half(0) ? __low2half(b) : __low2half(a); - half result_high = __high2half(mask) == __half(0) ? __high2half(b) : __high2half(a); + half mask_low = __low2half(mask); + half mask_high = __high2half(mask); + half result_low = mask_low == half(0) ? __low2half(b) : __low2half(a); + half result_high = mask_high == half(0) ? __high2half(b) : __high2half(a); return __halves2half2(result_low, result_high); } |