昇腾社区首页
中文
注册

函数:allocator_set_alloc_advise_func_to_desc

产品支持情况

产品

是否支持

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

Atlas A2 训练系列产品/Atlas 800I A2 推理产品/A200I A2 Box 异构组件

Atlas 训练系列产品

Atlas 推理系列产品

Atlas 200I/500 A2 推理产品

Atlas 200/300/500 推理产品

使用说明

c函数原型

aclError aclrtAllocatorSetAllocAdviseFuncToDesc(aclrtAllocatorDesc allocatorDesc, aclrtAllocatorAllocAdviseFunc func)

Python函数

ret = acl.rt.allocator_set_alloc_advise_func_to_desc(allocatorDesc, func)

函数功能

使用用户提供的Allocator场景下,设置"根据建议地址申请内存block"的回调函数,一般用于内存复用场景。

输入说明

  • allocatorDesc:int,Allocator描述符指针地址。需提前调用acl.rt.allocator_create_desc接口设置Allocator描述信息。
  • func:根据建议地址申请内存block的回调函数。

    回调函数定义如下:

    def allocator_alloc_advise_func(allocator, size, addr):
        pass

返回值说明

ret:int,返回0表示成功,返回其他值表示失败。