devtools/eigen
1
0
Fork 0
mirror of https://gitlab.com/libeigen/eigen.git synced 2026-04-10 11:34:33 +08:00
Code Issues Packages Projects Releases Wiki Activity

Added the ability to compute the absolute value of a half float

Browse Source
This commit is contained in:
Benoit Steiner
2016-02-21 20:24:11 +00:00
parent 203490017f
commit 95fceb6452
2 changed files with 49 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -52,9 +52,13 @@ __device__ half operator /= (half& a, const half& b) {
a = a / b;
return a;
}
__device__ half __shfl_xor(half a, int) {
assert(false && "tbd");
return a;
namespace std {
__device__ half abs(const half& a) {
half result;
result.x = a.x & 0x7FFF;
return result;
}
}
namespace Eigen {
@@ -214,8 +218,9 @@ template<> EIGEN_DEVICE_FUNC inline half predux_mul<half2>(const half2& a) {
}
template<> EIGEN_DEVICE_FUNC inline half2 pabs<half2>(const half2& a) {
assert(false && "tbd");
return half2();
half2 result;
result.x = a.x & 0x7FFF7FFF;
return result;
}