GitHub-Proxy/eigen
1
0
Fork 0
mirror of https://gitlab.com/libeigen/eigen.git synced 2025-08-12 11:49:02 +08:00
Code Issues Packages Projects Releases Wiki Activity

Print some information to stderr when a CUDA kernel fails

Browse Source
This commit is contained in:
Benoit Steiner 2016-02-27 20:42:57 +00:00
parent 1031b31571
commit 609b3337a7
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
unsupported/Eigen/CXX11/src/Tensor/TensorDeviceCuda.h
View File

@ -211,8 +211,12 @@ struct GpuDevice {
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE void synchronize() const { EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE void synchronize() const {
#if defined(__CUDACC__) && !defined(__CUDA_ARCH__) #if defined(__CUDACC__) && !defined(__CUDA_ARCH__)
cudaError_t err = cudaStreamSynchronize(stream_->stream()); cudaError_t err = cudaStreamSynchronize(stream_->stream());
EIGEN_UNUSED_VARIABLE(err) if (err != cudaSuccess) {
assert(err == cudaSuccess); std::cerr << "Error detected in CUDA stream: "
<< cudaGetErrorString(err)
<< std::endl;
assert(err == cudaSuccess);
}
#else #else
assert(false && "The default device should be used instead to generate kernel code"); assert(false && "The default device should be used instead to generate kernel code");
#endif #endif