include <iostream> in the tensor header since we now use it to better report cuda initialization errors

This commit is contained in:
Benoit Steiner 2016-02-22 13:59:03 -08:00
parent 257b640463
commit 5cd00068c0

View File

@ -58,6 +58,7 @@ typedef unsigned __int64 uint64_t;
#endif
#ifdef EIGEN_USE_GPU
#include <iostream>
#include <cuda_runtime.h>
#if defined(__CUDACC__)
#include <curand_kernel.h>