Use getter

chainer · Oct 28, 2017 · bb0b47d · bb0b47d
1 parent cd23288
commit bb0b47d
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/chainer/cuda.py b/chainer/cuda.py
@@ -127,8 +127,9 @@ def __ne__(self, other):
 # Global states
 # ------------------------------------------------------------------------------
 if available:
-    memory_pool = cupy.default_memory_pool
-    pinned_memory_pool = cupy.default_pinned_memory_pool
+    # This is for backward compatibility
+    memory_pool = cupy.get_default_memory_pool()
+    pinned_memory_pool = cupy.get_default_pinned_memory_pool()
 
 
 if six.PY2: