aclrtMallocForTaskScheduler

产品支持情况

产品

是否支持

Atlas A2 训练系列产品/Atlas 800I A2 推理产品/A200I A2 Box 异构组件

Atlas 200I/500 A2 推理产品

Atlas 推理系列产品

Atlas 训练系列产品

Atlas 200/300/500 推理产品

功能说明

申请昇腾AI处理器上Task调度器可使用的内存。

图模式下有部分算子需要使用该类型的内存。

函数原型

aclError aclrtMallocForTaskScheduler(void **devPtr, size_t size, aclrtMemMallocPolicy policy, aclrtMallocConfig *cfg)

参数说明

参数名

输入/输出

说明

devPtr

输出

“Device上已分配内存的指针”的指针。

size

输入

申请内存的大小,单位Byte。

size不能为0。

policy

输入

内存分配规则。

cfg

输入

内存配置信息。

不指定配置时,此处可传NULL。

返回值说明

返回0表示成功,返回其他值表示失败,请参见aclError