函数:malloc_for_task_scheduler
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)  | 
函数功能  | 
申请昇腾AI处理器上Task调度器可使用的内存。 图模式下有部分算子需要使用该类型的内存。  | 
输入说明  | 
size:int,申请内存的大小,单位Byte。size不能为0。 policy:int,内存分配规则。 若配置的内存分配规则超出aclrtMemMallocPolicy取值范围,size≥2M时,按大页申请内存,否则按普通页申请内存。 cfg:dict,内存配置信息。不指定配置时,此处可传空字典,具体请参见aclrtMallocConfig。  | 
输出说明  | 
dev_ptr:int,指向Device上已分配内存的指针地址。  | 
返回值说明  | 
ret:int,返回0表示成功,返回其他值表示失败,请参见aclError。  | 
父主题: 内存管理