Merge pull request #108 from drnikolaev/nvcaffe-0.14-orig

200169109 fix
NVIDIA · Jan 29, 2016 · 3ba0387 · 3ba0387
2 parents c887de9 + b0074e1
commit 3ba0387
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/caffe/util/gpu_memory.cpp b/src/caffe/util/gpu_memory.cpp
@@ -88,6 +88,8 @@ namespace caffe {
           }
           // retry once
           CUDA_CHECK(cubAlloc->DeviceAllocate(ptr, size, stream));
+          // If retry succeeds we need to clean up last error:
+          cudaGetLastError();
         }
       break;
     default: