昇腾社区首页
中文
注册

函数:malloc_with_cfg

C函数原型

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

Python函数

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

函数功能

在Device上分配size大小的线性内存,并通过dev_ptr返回已分配内存的指针。与acl.rt.malloc接口相比,本接口在申请内存时,还可以指定内存相关的配置信息。

输入说明

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

policy:int,内存分配规则。

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

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

输出说明

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

返回值说明

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