昇腾社区首页
中文
注册
开发者
下载

函数:register_cpu_func

产品支持情况

产品

是否支持

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

Atlas A2 训练系列产品/Atlas 800I A2 推理产品/A200I A2 Box 异构组件

Atlas 训练系列产品

Atlas 推理系列产品

Atlas 200I/500 A2 推理产品

功能说明

若使用acl.rt.binary_load_from_data接口加载AI CPU算子二进制数据,还需配合使用本接口注册AI CPU算子信息,得到对应的func_handle。本接口只用于AI CPU算子,其它算子会返回报错ACL_ERROR_RT_PARAM_INVALID。

函数原型

  • C函数原型
    1
    aclError aclrtRegisterCpuFunc(const aclrtBinHandle handle, const char *funcName, const char *kernelName, aclrtFuncHandle *funcHandle)
    
  • python函数
    1
    func_handle, ret = acl.rt.register_cpu_func(bin_handle, func_name, kernel_name)
    

参数说明

参数名

说明

bin_handle

int,算子二进制句柄。调用acl.rt.binary_load_from_data接口获取算子二进制句柄,再将其作为入参传入本接口。

func_name

str,执行AI CPU算子的入口函数。不能为空。

kernel_name

str,AI CPU算子的opType。不能为空。

返回值说明

返回值

说明

func_handle

int,核函数句柄。

ret

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