函数:register_cpu_func
产品支持情况
产品 |
是否支持 |
---|---|
√ |
|
√ |
|
√ |
|
√ |
|
√ |
功能说明
若使用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表示成功,返回其他值表示失败。 |
父主题: Kernel加载与执行