From 91f824dfb59883fad0413b113d5a1eba225762d7 Mon Sep 17 00:00:00 2001
From: Leo Fang <leofang@bnl.gov>
Date: Mon, 21 Sep 2020 16:35:14 -0400
Subject: [PATCH] fix managed memory leak

---
 cupy/cuda/memory.pyx | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git a/cupy/cuda/memory.pyx b/cupy/cuda/memory.pyx
index ed5be4b642c..1ae5df0addc 100644
--- a/cupy/cuda/memory.pyx
+++ b/cupy/cuda/memory.pyx
@@ -175,6 +175,11 @@ cdef class ManagedMemory(BaseMemory):
         """
         runtime.memAdvise(self.ptr, self.size, advise, device.id)
 
+    def __dealloc__(self):
+        if self.ptr:
+            syncdetect._declare_synchronize()
+            runtime.free(self.ptr)
+
 
 cdef set _peer_access_checked = set()
 
@@ -1360,9 +1365,9 @@ ctypedef void*(*malloc_func_type)(void*, size_t, int)
 ctypedef void(*free_func_type)(void*, void*, int)
 
 
-cdef size_t _call_malloc(
+cdef intptr_t _call_malloc(
         intptr_t param, intptr_t malloc_func, Py_ssize_t size, int device_id):
-    return <size_t>(
+    return <intptr_t>(
         (<malloc_func_type>malloc_func)(<void*>param, size, device_id))