From 6cd7b9ea6b401d84da5114041321ef388dc92092 Mon Sep 17 00:00:00 2001
From: Gael Guennebaud <g.gael@free.fr>
Date: Mon, 29 Aug 2016 11:06:08 +0200
Subject: [PATCH] Fix compilation with cuda 8

---
 test/cuda_basic.cu                               | 11 ++++++++++-
 unsupported/test/cxx11_tensor_of_float16_cuda.cu |  2 +-
 2 files changed, 11 insertions(+), 2 deletions(-)

diff --git a/test/cuda_basic.cu b/test/cuda_basic.cu
index b36ed888d..3cf37d221 100644
--- a/test/cuda_basic.cu
+++ b/test/cuda_basic.cu
@@ -1,4 +1,11 @@
-
+// This file is part of Eigen, a lightweight C++ template library
+// for linear algebra.
+//
+// Copyright (C) 2015-2016 Gael Guennebaud <gael.guennebaud@inria.fr>
+//
+// This Source Code Form is subject to the terms of the Mozilla
+// Public License v. 2.0. If a copy of the MPL was not distributed
+// with this file, You can obtain one at http://mozilla.org/MPL/2.0/.
 
 // workaround issue between gcc >= 4.7 and cuda 5.5
 #if (defined __GNUC__) && (__GNUC__>4 || __GNUC_MINOR__>=7)
@@ -12,6 +19,8 @@
 #define EIGEN_DEFAULT_DENSE_INDEX_TYPE int
 
 #include <math_constants.h>
+#include <cuda.h>
+#include <cuda_fp16.h>
 #include "main.h"
 #include "cuda_common.h"
 
diff --git a/unsupported/test/cxx11_tensor_of_float16_cuda.cu b/unsupported/test/cxx11_tensor_of_float16_cuda.cu
index 853b8d249..a6375d34a 100644
--- a/unsupported/test/cxx11_tensor_of_float16_cuda.cu
+++ b/unsupported/test/cxx11_tensor_of_float16_cuda.cu
@@ -13,7 +13,7 @@
 #define EIGEN_DEFAULT_DENSE_INDEX_TYPE int
 #define EIGEN_USE_GPU
 
-
+#include <cuda_fp16.h>
 #include "main.h"
 #include <unsupported/Eigen/CXX11/Tensor>