aclrtAllocatorGetByStream

Atlas A3 训练系列产品/Atlas A3 推理系列产品

√

Atlas A2 训练系列产品/Atlas A2 推理系列产品

√

Atlas 200I/500 A2 推理产品

x

Atlas 推理系列产品

√

Atlas 训练系列产品

√

stream

输入

注册的类型，按照不同的子模块区分。

allocatorDesc

输出

Allocator描述符指针。

allocator

输出

用户提供的Allocator对象指针。

allocFunc

输出

申请内存block的回调函数。

回调函数定义如下：

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

freeFunc

输出

释放内存block的回调函数。

回调函数定义如下：

typedef void (*aclrtAllocatorFreeFunc)(aclrtAllocator allocator, aclrtAllocatorBlock block);

allocAdviseFunc

输出

根据建议地址申请内存block的回调函数。

回调函数定义如下：

typedef void *(*aclrtAllocatorAllocAdviseFunc)(aclrtAllocator allocator, size_t size, aclrtAllocatorAddr addr);

getAddrFromBlockFunc

输出

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

回调函数定义如下：

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