frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

context gpu: better error catching

Browse Source
This commit is contained in:
Yangqing Jia
2015-12-14 13:59:28 -08:00
parent 73f3daf736
commit 3b0cc79465
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
caffe2/core/context_gpu.h
View File

@ -56,9 +56,11 @@ class CUDAContext {
}
inline bool FinishDeviceComputation() {
cudaError_t error = cudaStreamSynchronize(cuda_stream_);
cudaStreamSynchronize(cuda_stream_);
cudaError_t error = cudaGetLastError();
if (error != cudaSuccess) {
CAFFE_LOG_ERROR << cudaGetErrorString(error);
CAFFE_LOG_ERROR << "Encountered CUDA error: "
<< cudaGetErrorString(error);
return false;
}
return true;