下载
中文
注册
我要评分
文档获取效率
文档正确性
内容完整性
文档易理解
在线提单
论坛求助
昇腾小AI

函数:malloc_cached

C函数原型

aclError aclrtMallocCached(void **devPtr, size_t size, aclrtMemMallocPolicy policy)

Python函数

dev_ptr, ret = acl.rt.malloc_cached(size, policy)

函数功能

申请Device上的内存,该接口在任何场景下申请的内存都是支持cache缓存。在Device上申请size大小的线性内存,通过dev_ptr返回已分配内存的指针地址。

使用acl.rt.malloc_cached接口申请的内存与使用acl.rt.malloc接口申请的内存是等价的,都支持cache缓存,不需要用户处理cpu与npu之间的cache一致性。

调用媒体数据处理的接口前,若需要申请Device上的内存存放输入或输出数据,需调用acl.media.dvpp_malloc申请内存。

输入说明

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

policy:int,内存分配规则。

  • 0:ACL_MEM_MALLOC_HUGE_FIRST,当申请的内存小于等于1M时,即使使用该内存分配规则,也是申请普通页的内存。当申请的内存大于1M时,优先申请大页内存,如果大页内存不够,则使用普通页的内存。
  • 1:ACL_MEM_MALLOC_HUGE_ONLY,仅申请大页,如果大页内存不够,则返回错误。
  • 2:ACL_MEM_MALLOC_NORMAL_ONLY,仅申请普通页。
  • 3:ACL_MEM_MALLOC_HUGE_FIRST_P2P, 仅Device之间内存复制场景下申请内存时使用该选项,表示优先申请大页内存,如果大页内存不够,则使用普通页的内存。预留选项。
  • 4:ACL_MEM_MALLOC_HUGE_ONLY_P2P,仅Device之间内存复制场景下申请内存时使用该选项,仅申请大页内存,如果大页内存不够,则返回错误。预留选项。
  • 5:ACL_MEM_MALLOC_NORMAL_ONLY_P2P,仅Device之间内存复制场景下申请内存时使用该选项,仅申请普通页的内存。预留选项。

返回值说明

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

ret:int,错误码。

  • 返回0表示成功。
  • 返回其它值表示失败。

约束说明

其它约束与acl.rt.malloc接口相同。

搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词