函数：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 推理产品

√

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

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

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

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

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

policy：int，内存分配规则。

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

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

dev_ptr：int，指向Device上已分配内存的指针地址。

ret：int，返回0表示成功，返回其他值表示失败，请参见aclError。