From 17b9fbed34cefe08b4f63dbe0734e12311eb8669 Mon Sep 17 00:00:00 2001
From: Benoit Steiner <benoit.steiner.goog@gmail.com>
Date: Fri, 19 Feb 2016 06:16:07 +0000
Subject: Added preliminary support for half floats on CUDA GPU. For now we can
 simply convert floats into half floats and vice versa

---
 Eigen/Core | 3 +++
 1 file changed, 3 insertions(+)

(limited to 'Eigen/Core')

diff --git a/Eigen/Core b/Eigen/Core
index 63602f4c3..17f864084 100644
--- a/Eigen/Core
+++ b/Eigen/Core
@@ -200,6 +200,7 @@
 #if defined __CUDACC__
   #define EIGEN_VECTORIZE_CUDA
   #include <vector_types.h>
+  #include <cuda_fp16.h>
 #endif
 
 #if (defined _OPENMP) && (!defined EIGEN_DONT_PARALLELIZE)
@@ -329,7 +330,9 @@ using std::ptrdiff_t;
 
 #if defined EIGEN_VECTORIZE_CUDA
   #include "src/Core/arch/CUDA/PacketMath.h"
+  #include "src/Core/arch/CUDA/PacketMathHalf.h"
   #include "src/Core/arch/CUDA/MathFunctions.h"
+  #include "src/Core/arch/CUDA/TypeCasting.h"
 #endif
 
 #include "src/Core/arch/Default/Settings.h"
-- 
cgit v1.2.3