Try Try to fix some memory leaks.

Author: notoraptor

pygpu/gpuarray.pyx

@@ -1428,6 +1428,7 @@ cdef GpuArray pygpu_reshape(GpuArray a, unsigned int nd, const size_t *newdims,
             tot *= newdims[i]
     cdims = <size_t *>calloc(nd, sizeof(size_t))
     if cdims == NULL:
+        free(cdims)
         raise MemoryError, "could not allocate cdims"
 
     cdef size_t d
@@ -1437,10 +1438,12 @@ cdef GpuArray pygpu_reshape(GpuArray a, unsigned int nd, const size_t *newdims,
             d = a.size // tot
 
             if d * tot != a.size:
+                free(cdims)
                 raise GpuArrayException, "..."
         cdims[i] = d
 
     array_reshape(res, a, nd, cdims, ord, nocopy)
+    free(cdims)
     return res
 
 

src/cache/disk.c

@@ -216,8 +216,15 @@ static int key_path(disk_cache *c, const cache_key_t key, char *out) {
   unsigned char hash[64];
   int i;
 
-  if (c->kwrite(&kb, key)) return -1;
-  if (Skein_512((unsigned char *)kb.s, kb.l, hash)) return -1;
+  if (c->kwrite(&kb, key)) {
+    strb_clear(&kb);
+    return -1;
+  }
+  if (Skein_512((unsigned char *)kb.s, kb.l, hash)) {
+    strb_clear(&kb);
+    return -1;
+  }
+  strb_clear(&kb);
   if (snprintf(out, 10, "%02x%02x/%02x%02x",
                hash[0], hash[1], hash[2], hash[3]) != 9)
     return -1;

src/gpuarray_elemwise.c

@@ -738,18 +738,24 @@ GpuElemwise *GpuElemwise_new(gpucontext *ctx,
 
 void GpuElemwise_free(GpuElemwise *ge) {
   unsigned int i;
-  for (i = 0; i < ge->nd; i++) {
-    if (k_initialized(&ge->k_basic_32[i]))
-      GpuKernel_clear(&ge->k_basic_32[i]);
-    if (k_initialized(&ge->k_basic[i]))
-      GpuKernel_clear(&ge->k_basic[i]);
-  }
+  if (ge->k_basic_32 != NULL)
+    for (i = 0; i < ge->nd; i++) {
+      if (k_initialized(&ge->k_basic_32[i]))
+        GpuKernel_clear(&ge->k_basic_32[i]);
+    }
+  if (ge->k_basic != NULL)
+    for (i = 0; i < ge->nd; i++) {
+      if (k_initialized(&ge->k_basic[i]))
+        GpuKernel_clear(&ge->k_basic[i]);
+    }
   if (ge->strides != NULL)
     for (i = 0; i < ge->narray; i++) {
       free(ge->strides[i]);
     }
   if (k_initialized(&ge->k_contig))
     GpuKernel_clear(&ge->k_contig);
+  free(ge->k_basic_32);
+  free(ge->k_basic);
   free_args(ge->n, ge->args);
   free((void *)ge->preamble);
   free((void *)ge->expr);