aclrtAllocatorSetGetAddrFromBlockFuncToDesc

产品	是否支持
Atlas A2 训练系列产品/Atlas 800I A2 推理产品/A200I A2 Box 异构组件	√
Atlas 200I/500 A2 推理产品	√
Atlas 推理系列产品	√
Atlas 训练系列产品	√
Atlas 200/300/500 推理产品	√

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

aclError aclrtAllocatorSetGetAddrFromBlockFuncToDesc(aclrtAllocatorDesc allocatorDesc, aclrtAllocatorGetAddrFromBlockFunc func)

参数名	输入/输出	说明
allocatorDesc	输入	Allocator描述符指针。需提前调用aclrtAllocatorCreateDesc接口设置Allocator描述信息。
func	输入	根据申请来的block获取device内存地址的回调函数。回调函数定义如下： typedef void (aclrtAllocatorGetAddrFromBlockFunc)(aclrtAllocatorBlock block); aclrtAllocatorBlock的定义如下： typedef void *aclrtAllocatorBlock;

参数名

输入/输出

说明

allocatorDesc

输入

Allocator描述符指针。

需提前调用aclrtAllocatorCreateDesc接口设置Allocator描述信息。

func

输入

根据申请来的block获取device内存地址的回调函数。

回调函数定义如下：

typedef void *(*aclrtAllocatorGetAddrFromBlockFunc)(aclrtAllocatorBlock block);

aclrtAllocatorBlock的定义如下：

typedef void *aclrtAllocatorBlock;

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

父主题： aclrtAllocatorDesc