cupy.cuda.set_pinned_memory_allocator¶
-
cupy.cuda.
set_pinned_memory_allocator
(allocator=None)¶ Sets the current allocator for the pinned memory.
Parameters: allocator (function) – CuPy pinned memory allocator. It must have the same interface as the cupy.cuda.alloc_pinned_memory()
function, which takes the buffer size as an argument and returns the device buffer of that size. WhenNone
is specified, raw memory allocator is used (i.e., memory pool is disabled).