diff options
author | Benoit Steiner <benoit.steiner.goog@gmail.com> | 2016-09-26 11:00:32 -0700 |
---|---|---|
committer | Benoit Steiner <benoit.steiner.goog@gmail.com> | 2016-09-26 11:00:32 -0700 |
commit | 6565f8d60fc0cadd1935ab2dee522ec52829e615 (patch) | |
tree | 0927289b8ed09af5dad4f46ea87c31997ed392e3 /unsupported/Eigen/CXX11/Tensor | |
parent | 48dfe98abd00f2cb9b62d157f805f69d01b7892b (diff) |
Made the initialization of a CUDA device thread safe.
Diffstat (limited to 'unsupported/Eigen/CXX11/Tensor')
-rw-r--r-- | unsupported/Eigen/CXX11/Tensor | 4 |
1 files changed, 4 insertions, 0 deletions
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" |