mindspore.runtime.MemPool

class mindspore.runtime.MemPool(allocator)[source]

A MemPool warp a mindspore.runtime.PluggableAllocator, and pass it to mindspore.runtime.use_mem_pool().

Parameters: allocator (mindspore.runtime.PluggableAllocator) – a mindspore.runtime.PluggableAllocator that can be used to define how memory gets allocated and freed in the pool.

Supported Platforms:: Ascend