Fixed the +=, -=, *= and /= operators to return a reference

This commit is contained in:
Benoit Steiner 2016-03-10 15:17:44 -08:00
parent 86d45a3c83
commit 456e038a4e

View File

@ -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;
}