diff options
author | Benoit Steiner <benoit.steiner.goog@gmail.com> | 2016-11-01 12:08:09 -0700 |
---|---|---|
committer | Benoit Steiner <benoit.steiner.goog@gmail.com> | 2016-11-01 12:08:09 -0700 |
commit | 7a0e96b80dc622e3802ec158a12c63b728d7fce9 (patch) | |
tree | c774646f5b4ddb8fc9e51c021fa1db4bc5a7de91 /Eigen/src/Core/arch/CUDA | |
parent | 0a9ad6fc72f83ac0d3e4fa8d0fa6b339969f341a (diff) |
Gate the code that refers to cuda fp16 primitives more thoroughly
Diffstat (limited to 'Eigen/src/Core/arch/CUDA')
-rw-r--r-- | Eigen/src/Core/arch/CUDA/PacketMathHalf.h | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/Eigen/src/Core/arch/CUDA/PacketMathHalf.h b/Eigen/src/Core/arch/CUDA/PacketMathHalf.h index 9dd89e07f..de905bbf4 100644 --- a/Eigen/src/Core/arch/CUDA/PacketMathHalf.h +++ b/Eigen/src/Core/arch/CUDA/PacketMathHalf.h @@ -15,7 +15,7 @@ namespace Eigen { namespace internal { // Most of the following operations require arch >= 3.0 -#if defined(EIGEN_HAS_CUDA_FP16) && defined(__CUDA_ARCH__) && __CUDA_ARCH__ >= 300 +#if defined(EIGEN_HAS_CUDA_FP16) && defined(__CUDACC__) && defined(__CUDA_ARCH__) && __CUDA_ARCH__ >= 300 template<> struct is_arithmetic<half2> { enum { value = true }; }; |