GitHub-Proxy/eigen
1
0
Fork 0
mirror of https://gitlab.com/libeigen/eigen.git synced 2025-08-14 12:46:00 +08:00
Code Issues Packages Projects Releases Wiki Activity

Added support for operators +=, -=, *= and /= on CUDA half floats

Browse Source
This commit is contained in:
Benoit Steiner 2016-02-19 15:57:26 +00:00
parent dc26459b99
commit f7cb755299
1 changed files with 16 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -39,6 +39,22 @@ __device__ half operator / (const half& a, const half& b) {
__device__ half operator - (const half& a) { __device__ half operator - (const half& a) {
return __hneg(a); return __hneg(a);
} }
__device__ half operator += (half& a, const half& b) {
a = __hadd(a, b);
return a;
}
__device__ half operator *= (half& a, const half& b) {
a = __hmul(a, b);
return a;
}
__device__ half operator -= (half& a, const half& b) {
a = __hsub(a, b);
return a;
}
__device__ half operator /= (half& a, const half& b) {
assert(false && "tbd");
return a;
}
template<> struct is_arithmetic<half2> { enum { value = true }; }; template<> struct is_arithmetic<half2> { enum { value = true }; };