---
title: 函数：allocator_register
description: "| 产品 | 是否支持 |"
url: https://www.hiascend.com/document/detail/zh/canncommercial/latest/API/runtimeapi/aclpythondevg_01_0981.html
sourcePath: /source/zh/canncommercial/900/API/runtimeapi/aclpythondevg_01_0981.html
indexId: e74154be28405095d1e72a674612193d37550dbf2282f582d3b47425acc7971d71
---
# 函数：allocator_register

#### 产品支持情况

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


#### 功能说明

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


#### 函数原型

- C函数原型
  1 aclError aclrtAllocatorRegister(aclrtStream stream, aclrtAllocatorDesc allocatorDesc )

- python函数
  1 ret = acl.rt.allocator_register(stream, allocatorDesc)


#### 参数说明

| 参数名 | 说明 |
| --- | --- |
| stream | int，该Allocator需要注册的Stream。传入的stream参数值不能为NULL，否则返回报错。 |
| allocatorDesc | int，Allocator描述符指针地址。 |


#### 返回值说明

| 返回值 | 说明 |
| --- | --- |
| ret | int，错误码，返回0表示成功，返回其它值表示失败。 |


#### 约束说明

- Atlas 200I/500 A2推理产品，不支持该接口。
- 当前仅支持在单算子模型执行、动态shape模型推理场景下使用本接口。
  单算子模型场景下，需在算子执行接口（例如：acl.op.execute_v2(https://www.hiascend.comdocument/detail/zh/canncommercial/900/API/ascendgraphapi/aclpythondevg_01_0146.html)等）之前调用本接口。

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

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


#### 资源参考

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