diff options
author | Benoit Steiner <benoit.steiner.goog@gmail.com> | 2016-05-10 17:04:01 -0700 |
---|---|---|
committer | Benoit Steiner <benoit.steiner.goog@gmail.com> | 2016-05-10 17:04:01 -0700 |
commit | 4ede059de17f54b5c29b27e86486016caef0b795 (patch) | |
tree | 752f18086942c5684ababfee593b2330c8649d5c /unsupported/Eigen/CXX11/src/Tensor/TensorReductionCuda.h | |
parent | bf185c3c28073c5fd2c2e88cfcd2438bf1074677 (diff) |
Properly gate the use of half2.
Diffstat (limited to 'unsupported/Eigen/CXX11/src/Tensor/TensorReductionCuda.h')
-rw-r--r-- | unsupported/Eigen/CXX11/src/Tensor/TensorReductionCuda.h | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/unsupported/Eigen/CXX11/src/Tensor/TensorReductionCuda.h b/unsupported/Eigen/CXX11/src/Tensor/TensorReductionCuda.h index afa1a2697..6db9e63c8 100644 --- a/unsupported/Eigen/CXX11/src/Tensor/TensorReductionCuda.h +++ b/unsupported/Eigen/CXX11/src/Tensor/TensorReductionCuda.h @@ -68,6 +68,7 @@ __device__ EIGEN_ALWAYS_INLINE void atomicReduce(T* output, T accum, R& reducer) } +#ifdef EIGEN_HAS_CUDA_FP16 template <template <typename T> class R> __device__ inline void atomicReduce(half2* output, half2 accum, R<half>& reducer) { #if __CUDA_ARCH__ >= 300 @@ -90,6 +91,7 @@ __device__ inline void atomicReduce(half2* output, half2 accum, R<half>& reducer assert(0 && "Shouldn't be called on unsupported device"); #endif } +#endif template <> __device__ inline void atomicReduce(float* output, float accum, SumReducer<float>&) { |