mirror of
https://gitlab.com/libeigen/eigen.git
synced 2025-04-30 15:54:13 +08:00
include <iostream> in the tensor header since we now use it to better report cuda initialization errors
This commit is contained in:
parent
257b640463
commit
5cd00068c0
@ -58,6 +58,7 @@ typedef unsigned __int64 uint64_t;
|
|||||||
#endif
|
#endif
|
||||||
|
|
||||||
#ifdef EIGEN_USE_GPU
|
#ifdef EIGEN_USE_GPU
|
||||||
|
#include <iostream>
|
||||||
#include <cuda_runtime.h>
|
#include <cuda_runtime.h>
|
||||||
#if defined(__CUDACC__)
|
#if defined(__CUDACC__)
|
||||||
#include <curand_kernel.h>
|
#include <curand_kernel.h>
|
||||||
|
Loading…
x
Reference in New Issue
Block a user