devtools/eigen
1
0
Fork 0
mirror of https://gitlab.com/libeigen/eigen.git synced 2026-04-10 11:34:33 +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
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