mindspore.runtime.MemPool

View Source On Gitee
class mindspore.runtime.MemPool(allocator: PluggableAllocator)[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