aclrtAllocatorSetAllocFuncToDesc

函数功能

使用用户提供的Allocator场景下,设置"申请内存block"的回调函数。

函数原型

aclError aclrtAllocatorSetAllocFuncToDesc(aclrtAllocatorDesc allocatorDesc, aclrtAllocatorAllocFunc func)

参数说明

参数名

输入/输出

说明

allocatorDesc

输入

Allocator描述符指针。

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

func

输入

申请内存block的回调函数。

回调函数定义如下:

typedef void *(*aclrtAllocatorAllocFunc)(aclrtAllocator allocator, size_t size);

aclrtAllocator的定义如下:

typedef void *aclrtAllocator;

返回值说明

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