函数：allocator_set_get_addr_from_block_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 推理产品	√

使用用户提供的Allocator场景下，设置"根据申请来的block获取device内存地址"的回调函数。

C函数原型

aclError aclrtAllocatorSetGetAddrFromBlockFuncToDesc(aclrtAllocatorDesc allocatorDesc, aclrtAllocatorGetAddrFromBlockFunc func)

python函数

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

参数名	说明
allocatorDesc	int，Allocator描述符指针地址。需提前调用acl.rt.allocator_create_desc接口设置Allocator描述信息。
func	根据申请来的block获取device内存地址的回调函数。回调函数定义如下： def allocator_get_addr_from_block_func(block): pass

返回值	说明
ret	int，返回0表示成功，返回其他值表示失败。

父主题： aclrtAllocatorDesc