[发明专利]一种深度学习框架之中算子的部署方法、装置及电子设备

权利要求书

1.一种深度学习框架之中算子的部署方法，包括：

获取算子的源文件；

对所述算子的源文件进行编译以形成所述算子的动态链接库；

生成所述算子的动态链接库的前端系统的接口文件；

根据所述动态链接库和所述接口文件生成可安装库文件；

将所述可安装库文件安装至目标编程语言库；

获取所述动态链接库对应的待注入的函数代码；

将所述待注入的函数代码注入入口文件中，以生成所述动态链接库的目标调用函数代码；

将所述目标调用函数代码写入所述接口文件中；

其中，所述将所述待注入的函数代码注入入口文件中，以生成所述动态链接库的目标调用函数代码，包括：

从所述入口文件中获取所述动态链接库的默认调用函数代码；

将所述默认调用函数代码替换为所述待注入的函数代码，以生成所述动态链接库的目标调用函数代码。

2.根据权利要求1所述的部署方法，其中，所述对所述算子的源文件进行编译以形成所述算子的动态链接库，包括：

对所述算子的源文件进行头文件展开和宏替换，以生成待编译代码；

将所述待编译代码编译成汇编代码，并将所述汇编代码逐行转换成字节码，以生成目标文件；

对所述目标文件进行链接操作以获取链接后的目标文件，并根据所述链接后的目标文件生成所述动态链接库。

3.根据权利要求1所述的部署方法，其中，所述根据所述动态链接库和所述接口文件生成可安装库文件，包括：

按照预设打包策略对所述动态链接库和所述接口文件进行打包，以生成所述可安装库文件。

4.根据权利要求1所述的部署方法，其中，所述将所述可安装库文件安装至目标编程语言库，包括：

将所述可安装库文件安装至所述目标编程语言库中的目标目录下。

5.根据权利要求1所述的部署方法，其中，还包括：

获取目标算子的导入语句，其中，所述导入语句中携带所述目标算子的可安装库文件所在编程语言库对应的信息；

对所述导入语句进行解析，以确定所述目标算子的可安装库文件所在的目标子目录；

从所述目标子目录下读取所述目标算子的可安装库文件并进行导入。

6.一种深度学习框架之中算子的部署装置，包括：

第一获取模块，用于获取算子的源文件；

形成模块，用于对所述算子的源文件进行编译以形成所述算子的动态链接库；

第一生成模块，用于生成所述算子的动态链接库的前端系统的接口文件；

第二生成模块，用于根据所述动态链接库和所述接口文件生成可安装库文件；

安装模块，用于将所述可安装库文件安装至目标编程语言库；

第三获取模块，用于获取所述动态链接库对应的待注入的函数代码；

注入模块，用于将所述待注入的函数代码注入入口文件中，以生成所述动态链接库的目标调用函数代码；

写入模块，用于将所述目标调用函数代码写入所述接口文件中；

其中，所述写入模块，包括：

第一获取子模块，用于从所述入口文件中获取所述动态链接库的默认调用函数代码；

第五生成子模块，用于将所述默认调用函数代码替换为所述待注入的函数代码，以生成所述动态链接库的目标调用函数代码。

7.根据权利要求6所述的部署装置，其中，所述形成模块，包括：

第一生成子模块，用于对所述算子的源文件进行头文件展开和宏替换，以生成待编译代码；

第二生成子模块，用于将所述待编译代码编译成汇编代码，并将所述汇编代码逐行转换成字节码，以生成目标文件；

第三生成子模块，用于对所述目标文件进行链接操作以获取链接后的目标文件，并根据所述链接后的目标文件生成所述动态链接库。

8.根据权利要求6所述的部署装置，其中，所述第二生成模块，包括：

第四生成子模块，用于按照预设打包策略对所述动态链接库和所述接口文件进行打包，以生成所述可安装库文件。