aclrtAllocatorRegister

产品支持情况

功能说明

调用该接口注册用户提供的Allocator以及Allocator对应的回调函数，用于使用用户提供的Allocator。

函数原型

aclError aclrtAllocatorRegister(aclrtStream stream, aclrtAllocatorDesc allocatorDesc)

参数说明

返回值说明

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

约束说明

• 当前仅支持在单算子模型执行、动态shape模型推理场景下使用本接口。

  单算子模型场景下，需在算子执行接口（例如：aclopExecuteV2、aclopCompileAndExecuteV2等）之前调用本接口。

  动态shape模型推理场景，本接口需配合aclmdlExecuteAsync接口一起使用，且需在aclmdlExecuteAsync接口之前调用本接口。

• 调用本接口前，需要先调用aclrtAllocatorCreateDesc创建Allocator描述符，再分别调用aclrtAllocatorSetObjToDesc、aclrtAllocatorSetAllocFuncToDesc、aclrtAllocatorSetGetAddrFromBlockFuncToDesc、aclrtAllocatorSetFreeFuncToDesc设置Allocator对象及回调函数。Allocator描述符使用完成后，可调用aclrtAllocatorDestroyDesc接口销毁Allocator描述符。
• 对于同一条流，多次调用本接口，以最后一次注册为准。
• 对于不同流，如果用户使用同一个Allocator，不可以多条流并发执行，在执行下一条Stream前，需要对上一Stream做流同步。
• 将Allocator中的内存释放给操作系统前，需要先调用aclrtSynchronizeStream接口执行流同步，确保Stream中的任务已执行完成。

参考资源

接口调用示例，参见动态Shape输入（设置Shape范围）。