From 6565f8d60fc0cadd1935ab2dee522ec52829e615 Mon Sep 17 00:00:00 2001
From: Benoit Steiner <benoit.steiner.goog@gmail.com>
Date: Mon, 26 Sep 2016 11:00:32 -0700
Subject: Made the initialization of a CUDA device thread safe.

---
 unsupported/Eigen/CXX11/Tensor | 4 ++++
 1 file changed, 4 insertions(+)

(limited to 'unsupported/Eigen/CXX11/Tensor')

diff --git a/unsupported/Eigen/CXX11/Tensor b/unsupported/Eigen/CXX11/Tensor
index f7b94cee1..1d9f89864 100644
--- a/unsupported/Eigen/CXX11/Tensor
+++ b/unsupported/Eigen/CXX11/Tensor
@@ -64,6 +64,10 @@ typedef unsigned __int64 uint64_t;
 #if defined(__CUDACC__)
 #include <curand_kernel.h>
 #endif
+#if __cplusplus >= 201103L
+#include <atomic>
+#include <unistd.h>
+#endif
 #endif
 
 #include "src/Tensor/TensorMacros.h"
-- 
cgit v1.2.3