Use CuPy default memory pool

chainer · Oct 28, 2017 · cd23288 · cd23288
1 parent 7289901
commit cd23288
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/chainer/cuda.py b/chainer/cuda.py
@@ -127,10 +127,8 @@ def __ne__(self, other):
 # Global states
 # ------------------------------------------------------------------------------
 if available:
-    memory_pool = cuda.MemoryPool()
-    cuda.set_allocator(memory_pool.malloc)
-    pinned_memory_pool = cuda.PinnedMemoryPool()
-    cuda.set_pinned_memory_allocator(pinned_memory_pool.malloc)
+    memory_pool = cupy.default_memory_pool
+    pinned_memory_pool = cupy.default_pinned_memory_pool
 
 
 if six.PY2: