Only extract chosen algo after checking for success

theano/sandbox/cuda/dnn_fwd.c

@@ -83,8 +83,14 @@ APPLY_SPECIFIC(conv_fwd)(CudaNdarray *input, CudaNdarray *kerns,
                                                      requestedCount,
                                                      &count,
                                                      &choosen_algo_perf);
+          if (err != CUDNN_STATUS_SUCCESS) {
+            PyErr_Format(PyExc_RuntimeError,
+                         "GpuDnnConv: error selecting convolution algo: %s",
+                         cudnnGetErrorString(err));
+            return 1;
+          }
+
           chosen_algo = choosen_algo_perf.algo;
-          fprintf(stdout, "Choose algo %i\n", chosen_algo);
         }
         else
         {
@@ -97,7 +103,6 @@ APPLY_SPECIFIC(conv_fwd)(CudaNdarray *input, CudaNdarray *kerns,
                                                     CUDNN_CONVOLUTION_FWD_SPECIFY_WORKSPACE_LIMIT,
                                                     free,
                                                     &chosen_algo);
-        }
 
           if (err != CUDNN_STATUS_SUCCESS) {
             PyErr_Format(PyExc_RuntimeError,
@@ -105,6 +110,7 @@ APPLY_SPECIFIC(conv_fwd)(CudaNdarray *input, CudaNdarray *kerns,
                          cudnnGetErrorString(err));
             return 1;
           }
+        }
 
         // Store the shapes of the inputs and kernels as well as the chosen
         // algorithm for future use.