Avoid free_all_free

cupy · Jun 20, 2017 · 5ffc080 · 5ffc080
1 parent 8de94bc
commit 5ffc080
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/cupy/cuda/memory.pyx b/cupy/cuda/memory.pyx
@@ -361,7 +361,7 @@ cdef class SingleDeviceMemoryPool:
             except runtime.CUDARuntimeError as e:
                 if e.status != runtime.errorMemoryAllocation:
                     raise
-                self.free_all_free()
+                self.free_all_blocks()
                 try:
                     mem = self._alloc(size).mem
                 except runtime.CUDARuntimeError as e: