昇腾社区首页
中文
注册
开发者
下载

MallocAdvise

产品支持情况

产品

是否支持

Atlas A3 训练系列产品 / Atlas A3 推理系列产品

Atlas A2 训练系列产品 / Atlas A2 推理系列产品

Atlas 200I/500 A2 推理产品

Atlas 推理系列产品

Atlas 训练系列产品

头文件

/include/ge/ge_allocator.h

功能说明

在用户内存池中根据指定size大小申请device内存,建议申请的内存地址为addr。

函数原型

1
virtual MemBlock *MallocAdvise(size_t size, void *addr)

参数说明

参数名

输入/输出

描述

size

输入

指定需要申请内存大小。

addr

输入

建议申请的内存地址为addr。

返回值说明

返回MemBlock指针。

约束说明

虚函数需要用户实现,如若未实现,默认同Malloc功能相同。