diff --git a/Eigen/src/Core/arch/CUDA/PacketMathHalf.h b/Eigen/src/Core/arch/CUDA/PacketMathHalf.h index 720155ce1..9e3c51d49 100644 --- a/Eigen/src/Core/arch/CUDA/PacketMathHalf.h +++ b/Eigen/src/Core/arch/CUDA/PacketMathHalf.h @@ -38,19 +38,19 @@ __device__ half operator / (const half& a, const half& b) { __device__ half operator - (const half& a) { return __hneg(a); } -__device__ half operator += (half& a, const half& b) { +__device__ half& operator += (half& a, const half& b) { a = a + b; return a; } -__device__ half operator *= (half& a, const half& b) { +__device__ half& operator *= (half& a, const half& b) { a = a * b; return a; } -__device__ half operator -= (half& a, const half& b) { +__device__ half& operator -= (half& a, const half& b) { a = a - b; return a; } -__device__ half operator /= (half& a, const half& b) { +__device__ half& operator /= (half& a, const half& b) { a = a / b; return a; }