aclrtMalloc

产品支持情况

功能说明

在Device上分配size大小的线性内存，并通过*devPtr返回已分配内存的指针，且内存首地址64字节对齐。使用本接口申请的内存，需要通过aclrtFree接口或aclrtFreeWithDevSync接口释放内存。

函数原型

1

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

参数说明

返回值说明

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

约束说明

• 本接口分配的内存不会对内容初始化，建议在使用内存前先调用aclrtMemset接口先初始化内存，清除内存中的随机数。
• 本接口内部不会进行隐式的Device同步或流同步，如果申请内存成功或申请内存失败会立刻返回结果。
• policy处仅支持配置单个枚举项，不支持配置多个枚举项位或。
• 针对ACL_MEM_MALLOC_HUGE_FIRST_P2P、ACL_MEM_MALLOC_HUGE_ONLY_P2P内存分配规则，建议使用aclrtMallocWithCfg接口，否则可能存在性能问题或无法申请到内存。
• 频繁调用aclrtMalloc接口申请内存、调用aclrtFree接口释放内存，会损耗性能，建议用户提前做内存预先分配或二次管理，避免频繁申请/释放内存。
• 若用户需申请大块内存并自行划分、管理内存时，每段内存需同时满足以下需求，其中，len表示某段内存的大小，ALIGN_UP[len,k]表示向上按k字节对齐：((len-1)/k+1)*k：
  • 内存大小对齐规则，各产品型号有所不同：
    • 对于 Ascend 950PR / Ascend 950DT ，内存大小向上对齐成32整数倍（m=ALIGN_UP[len,32]字节）。
    • 对于以下产品型号，内存大小向上对齐成32整数倍+32字节（m=ALIGN_UP[len,32]+32字节）。

      Atlas A3 训练系列产品 / Atlas A3 推理系列产品

      Atlas A2 训练系列产品 / Atlas A2 推理系列产品

      Atlas 200I/500 A2 推理产品

      Atlas 推理系列产品

      Atlas 训练系列产品

  • 内存起始地址需满足64字节对齐（ALIGN_UP[m,64]）。

接口调用示例

接口调用示例，参见Device内存使用。