Merge pull request BVLC#56 from lukeyeager/nvidia/fix-destroy-cudnn

Fix cuDNN handle destruction error
thatguymike · Oct 30, 2015 · 9068a9e · 9068a9e
2 parents ae1ac60 + d52bc27
commit 9068a9e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/caffe/common.cpp b/src/caffe/common.cpp
@@ -170,7 +170,7 @@ void Caffe::SetDevice(const int device_id) {
   CURAND_CHECK(curandSetPseudoRandomGeneratorSeed(Get().curand_generator_,
       cluster_seedgen()));
 #ifdef USE_CUDNN
-  if (Get().cublas_handle_) CUDNN_CHECK(cudnnDestroy(Get().cudnn_handle_));
+  if (Get().cudnn_handle_) CUDNN_CHECK(cudnnDestroy(Get().cudnn_handle_));
   CUDNN_CHECK(cudnnCreate(&Get().cudnn_handle_));
 #endif
 }