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