aclrtAllocatorSetGetAddrFromBlockFuncToDesc

函数功能

使用用户提供的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;

返回值说明

返回0表示成功,返回其它值表示失败。