Made the initialization of a CUDA device thread safe.

This commit is contained in:
Benoit Steiner
2016-09-26 11:00:32 -07:00
parent 48dfe98abd
commit 6565f8d60f
2 changed files with 31 additions and 1 deletions

View File

@@ -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"