---
title: aclrtAllocatorRegister
description: "| 产品 | 是否支持 |"
url: https://www.hiascend.com/document/detail/zh/canncommercial/latest/API/runtimeapi/aclcppdevg_03_0110.html
sourcePath: /source/zh/canncommercial/900/API/runtimeapi/aclcppdevg_03_0110.html
indexId: 0df6857e8bf105d40898a2a45cbd6ea7585a4b0ca9ef41fc849661d3b3c435c868
---
# aclrtAllocatorRegister

#### 产品支持情况

| 产品 | 是否支持 |
| --- | --- |
| Atlas 350 加速卡 | √ |
| Atlas A3 训练系列产品 / Atlas A3 推理系列产品 | √ |
| Atlas A2 训练系列产品 / Atlas A2 推理系列产品 | √ |
| Atlas 200I/500 A2 推理产品 | ☓ |
| Atlas 推理系列产品 | √ |
| Atlas 训练系列产品 | √ |


#### 功能说明

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


#### 函数原型

```
aclError aclrtAllocatorRegister(aclrtStream stream, aclrtAllocatorDesc allocatorDesc)
```


#### 参数说明

| 参数名 | 输入/输出 | 说明 |
| --- | --- | --- |
| stream | 输入 | 该Allocator需要注册的Stream。类型定义请参见aclrtStream。 传入的stream参数值不能为NULL，否则返回报错。 |
| allocatorDesc | 输入 | Allocator描述符指针。类型定义请参见aclrtAllocatorDesc。 |


#### 返回值说明

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


#### 约束说明

- 当前仅支持在单算子模型执行、动态shape模型推理场景下使用本接口。
  单算子模型场景下，需在算子执行接口（例如aclopExecuteV2(https://www.hiascend.comdocument/detail/zh/canncommercial/900/API/ascendgraphapi/aclcppdevg_03_0250.html)）之前调用本接口。

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

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


#### 参考资源

接口调用示例，参见动态Shape输入（设置Shape范围）(https://www.hiascend.comdocument/detail/zh/canncommercial/900/programug/acldevg/aclcppdevg_000044.html)。