Add PyDecref(cols) when an error occurs

theano/sandbox/cuda/conv_gemm.cu

@@ -294,6 +294,7 @@ CudaNdarray* corrMM(CudaNdarray *const bottom,
                              "This could be a known bug in CUDA, please see the "
                              "GpuCorrMM() documentation.\n",
                              cudaGetErrorString(err));
+                Py_DECREF(col);
                 return NULL;
             }
             // Second, gemm
@@ -311,6 +312,7 @@ CudaNdarray* corrMM(CudaNdarray *const bottom,
                         "This could be a known bug in CUDA, please see the "
                         "GpuCorrMM() documentation.\n",
                         cublasGetErrorString(status));
+                Py_DECREF(col);
                 return NULL;
             }
         }
@@ -359,6 +361,7 @@ CudaNdarray* corrMM(CudaNdarray *const bottom,
                              "This could be a known bug in CUDA, please see the "
                              "GpuCorrMM() documentation.\n",
                              cudaGetErrorString(err));
+                Py_DECREF(col);
                 return NULL;
             }
             // Second, gemm
@@ -379,6 +382,7 @@ CudaNdarray* corrMM(CudaNdarray *const bottom,
                         "This could be a known bug in CUDA, please see the "
                         "GpuCorrMM() documentation.\n",
                         cublasGetErrorString(status));
+                Py_DECREF(col);
                 return NULL;
             }
         }
@@ -429,6 +433,7 @@ CudaNdarray* corrMM(CudaNdarray *const bottom,
                         "This could be a known bug in CUDA, please see the "
                         "GpuCorrMM() documentation.\n",
                         cublasGetErrorString(status));
+                Py_DECREF(col);
                 return NULL;
             }
             // col2im back to the data
@@ -441,6 +446,7 @@ CudaNdarray* corrMM(CudaNdarray *const bottom,
                              "This could be a known bug in CUDA, please see the "
                              "GpuCorrMM() documentation.\n",
                              cudaGetErrorString(err));
+                Py_DECREF(col);
                 return NULL;
             }
         }

theano/sandbox/cuda/corr3d_gemm.cu

@@ -356,6 +356,7 @@ CudaNdarray* corr3dMM(CudaNdarray *const bottom,
                        "This could be a known bug in CUDA, please see the "
                        "GpuCorr3dMM() documentation.\n",
                        cudaGetErrorString(err));
+          Py_DECREF(col);
           return 0;
         }
         // Second, gemm
@@ -374,6 +375,7 @@ CudaNdarray* corr3dMM(CudaNdarray *const bottom,
                        "This could be a known bug in CUDA, please see the "
                        "GpuCorr3dMM() documentation.\n",
                        cublasGetErrorString(status));
+          Py_DECREF(col);
           return 0;
         }
       }
@@ -401,6 +403,7 @@ CudaNdarray* corr3dMM(CudaNdarray *const bottom,
                       "This could be a known bug in CUDA, please see the "
                       "GpuCorr3dMM() documentation.\n",
                       cudaGetErrorString(err));
+         Py_DECREF(col);
          return 0;
        }
         // Second, gemm
@@ -422,6 +425,7 @@ CudaNdarray* corr3dMM(CudaNdarray *const bottom,
                        "This could be a known bug in CUDA, please see the "
                        "GpuCorr3dMM() documentation.\n",
                        cublasGetErrorString(status));
+          Py_DECREF(col);
           return 0;
         }
       }
@@ -450,6 +454,7 @@ CudaNdarray* corr3dMM(CudaNdarray *const bottom,
                        "This could be a known bug in CUDA, please see the "
                        "GpuCorr3dMM() documentation.\n",
                        cublasGetErrorString(status));
+          Py_DECREF(col);
           return 0;
         }
         // col2im3d back to the data
@@ -466,6 +471,7 @@ CudaNdarray* corr3dMM(CudaNdarray *const bottom,
                        "This could be a known bug in CUDA, please see the "
                        "GpuCorr3dMM() documentation.\n",
                        cudaGetErrorString(err));
+          Py_DECREF(col);
           return 0;
         }
       }