def __eq__(self, other):
return isinstance(other, DummyDeviceType)
def __ne__(self, other):
return not (self == other)
DummyDevice = DummyDeviceType()
# ------------------------------------------------------------------------------
# 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)
if six.PY2:
try:
from future.types.newint import newint as _newint
_integer_types = six.integer_types + (_newint, )
except ImportError:
_integer_types = six.integer_types
else:
_integer_types = six.integer_types
# ------------------------------------------------------------------------------
# Global states
# ------------------------------------------------------------------------------
"""
for arg in args:
if isinstance(
arg,
(ndarray, sparse.spmatrix, cupy.core.fusion.FusionVarPython)):
return _cupy
return numpy
fuse = cupy.core.fusion.fuse
disable_experimental_feature_warning = False
# set default allocator
_default_memory_pool = cuda.MemoryPool()
_default_pinned_memory_pool = cuda.PinnedMemoryPool()
cuda.set_allocator(_default_memory_pool.malloc)
cuda.set_pinned_memory_allocator(_default_pinned_memory_pool.malloc)
def get_default_memory_pool():
"""Returns CuPy default memory pool for GPU memory.
Returns:
cupy.cuda.MemoryPool: The memory pool object.
.. note::
If you want to disable memory pool, please use the following code.
>>> cupy.cuda.set_allocator(None)
for arg in args:
if isinstance(
arg,
(ndarray, sparse.spmatrix, cupy.core.fusion._FusionVarScalar,
cupy.core.fusion._FusionVarArray)):
return _cupy
return numpy
fuse = cupy.core.fusion.fuse
disable_experimental_feature_warning = False
# set default allocator
_default_memory_pool = cuda.MemoryPool(cupy.cuda.memory.malloc_managed)
_default_pinned_memory_pool = cuda.PinnedMemoryPool(
) #cupy.cuda.memory.malloc_managed)
_default_device_memory_pool = cuda.MemoryPool()
#print("it is OC-cupy")
cuda.set_allocator(_default_memory_pool.malloc)
cuda.set_pinned_memory_allocator(_default_pinned_memory_pool.malloc)
def get_default_memory_pool():
"""Returns CuPy default memory pool for GPU memory.
Returns:
cupy.cuda.MemoryPool: The memory pool object.
.. note::
If you want to disable memory pool, please use the following code.