昇腾社区首页
中文
注册

函数:allocator_register

C函数原型

aclError aclrtAllocatorRegister(aclrtStream stream, aclrtAllocatorDesc allocatorDesc )

Python函数

ret = acl.rt.allocator_register(stream, allocatorDesc)

函数功能

调用该接口注册用户提供的Allocator以及Allocator对应的回调函数,用于使用用户提供的Allocator。

输入说明

stream:该Allocator需要注册的Stream。传入的stream参数值不能为NULL,否则返回报错。

allocatorDesc:Allocator描述符指针地址。

返回值说明

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

约束说明

参考资源

接口调用示例,参见动态Shape输入(设置Shape范围)