mindspore.runtime.PluggableAllocator

class mindspore.runtime.PluggableAllocator(path_to_so_file: str, alloc_fn_name: str, free_fn_name: str)

通过ctypes接收一个.so文件，并动态加载其中的alloc和free函数。 该功能需要与 mindspore.runtime.MemPool 和 mindspore.runtime.use_mem_pool() 配合使用，以接管MindSpore内存池中的内存分配和释放操作。

警告

当前仅支持在Unix类操作系统上使用。

参数：

• path_to_so_file (str) - 文件系统中包含分配器函数的 .so 文件的路径。

• alloc_fn_name (str) - so文件中执行内存分配的函数名称。函数签名必须为： void* alloc_fn(size_t size, int device, aclrtStream stream); 。

• free_fn_name (str) - so文件中执行内存释放的函数名称。函数签名必须为： void free_fn(void* ptr, size_t size, aclrtStream stream); 。

支持平台：

Ascend