函数：malloc_for_task_scheduler

产品	是否支持
Atlas A3 训练系列产品/Atlas A3 推理系列产品	√
Atlas A2 训练系列产品/Atlas 800I A2 推理产品/A200I A2 Box 异构组件	√
Atlas 训练系列产品	√
Atlas 推理系列产品	√
Atlas 200I/500 A2 推理产品	√
Atlas 200/300/500 推理产品	√

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

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

C函数原型

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

python函数

dev_ptr, ret = acl.rt.malloc_for_task_scheduler(size, policy, cfg)

参数名	说明
size	int，申请内存的大小，单位Byte。size不能为0。
policy	int，内存分配规则。若配置的内存分配规则超出aclrtMemMallocPolicy取值范围，size≥2M时，按大页申请内存，否则按普通页申请内存。
cfg	dict，内存配置信息。不指定配置时，此处可传空字典，具体请参见aclrtMallocConfig。

参数名

说明

size

int，申请内存的大小，单位Byte。size不能为0。

policy

int，内存分配规则。

若配置的内存分配规则超出aclrtMemMallocPolicy取值范围，size≥2M时，按大页申请内存，否则按普通页申请内存。

cfg

dict，内存配置信息。不指定配置时，此处可传空字典，具体请参见aclrtMallocConfig。

返回值	说明
devPtr	int，指向Device上已分配内存的指针地址。
ret	int，错误码，返回0表示成功，返回其它值表示失败。

父主题： 内存管理