使用用户提供的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表示成功，返回其它值表示失败。