aclrtRegisterCpuFunc
产品支持情况
| 产品 | 是否支持 |
|---|---|
| Atlas A3 训练系列产品/Atlas A3 推理系列产品 | √ |
| Atlas A2 训练系列产品/Atlas A2 推理系列产品 | √ |
功能说明
若使用aclrtBinaryLoadFromData接口加载AI CPU算子二进制数据,还需配合使用本接口注册AI CPU算子信息,得到对应的funcHandle。
本接口只用于AI CPU算子,其它算子会返回报错ACL_ERROR_RT_PARAM_INVALID。
函数原型
aclError aclrtRegisterCpuFunc(const aclrtBinHandle handle, const char *funcName, const char *kernelName, aclrtFuncHandle *funcHandle)
参数说明
| 参数名 | 输入/输出 | 说明 |
|---|---|---|
| handle | 输入 | 算子二进制句柄。 调用aclrtBinaryLoadFromData接口获取算子二进制句柄,再将其作为入参传入本接口。 |
| funcName | 输入 | 执行AI CPU算子的入口函数。不能为空。 |
| kernelName | 输入 | AI CPU算子的opType。不能为空。 |
| funcHandle | 输出 | 函数句柄。 |
返回值说明
返回0表示成功,返回其他值表示失败,请参见aclError。