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. When None is specified, raw memory allocator is used (i.e., memory pool is disabled).