cupy.cuda.runtime.deviceSetMemPool

cupy.cuda.runtime.deviceSetMemPool(int device, intptr_t pool)

Set the current mempool on the current device to pool.

cupy.cuda.runtime.deviceGetMemPool cupy.cuda.runtime.memPoolTrimTo