mindspore.runtime.MemPool

class mindspore.runtime.MemPool(allocator)[源代码]

MemPool封装了一个 mindspore.runtime.PluggableAllocator ，并将其传递给 mindspore.runtime.use_mem_pool() 使用。

参数：

allocator (mindspore.runtime.PluggableAllocator) - 一个 mindspore.runtime.PluggableAllocator 对象，用于定义内存池中内存的分配和释放方式。

支持平台：

Ascend