【技术实现步骤摘要】
动态链接库文件的生成方法、算子的调用方法及装置
[0001]本公开涉及人工智能
,尤其涉及深度学习
,具体涉及一种动态链接库文件的生成方法、算子的调用方法及装置。
技术介绍
[0002]张量作为深度学习框架中的基本数据单元,在利用深度学习框架搭建人工智能模型时,通常是将张量作为输入,通过调用深度学习框架中的算子以完成特定的运算逻辑。
技术实现思路
[0003]本公开提供了一种动态链接库文件的生成方法、算子的调用方法及装置。
[0004]根据本公开的一方面,提供了一种动态链接库文件的生成方法,包括:获取可执行文件生成模板;从张量配置信息中得到算子核函数的名称信息;根据算子核函数的名称信息,从算子配置信息中得到算子核函数的参数信息;以及根据名称信息、参数信息和可执行文件生成模板,生成动态链接库文件,其中,动态链接库文件包括同一算子在不同运行场景的多个可执行文件。
[0005]根据本公开的另一方面,提供了一种深度学习框架算子的调用方法,包括:根据用于调用目标算子的源代码,确定目标运行场景和目标算子,其中,目标运行场景指示了目标算子的调用方式和目标算子的运行模式;根据目标运行场景和目标算子,从动态链接库文件中得到目标可执行文件,其中,动态链接库文件是利用上述动态链接库文件的生成方法生成的;以及通过运行目标可执行文件,以执行目标算子的运算逻辑。
[0006]根据本公开的另一方面,提供了一种动态链接库文件的生成装置,包括:获取模块、第一获得模块、第二获得模块和生成模块。获取模块 ...
【技术保护点】
【技术特征摘要】
1.一种动态链接库文件的生成方法,包括:获取可执行文件生成模板;从张量配置信息中得到算子核函数的名称信息;根据所述算子核函数的名称信息,从算子配置信息中得到算子核函数的参数信息;以及根据所述名称信息、所述参数信息和所述可执行文件生成模板,生成所述动态链接库文件,其中,所述动态链接库文件包括同一算子在不同运行场景的多个可执行文件。2.根据权利要求1所述的方法,其中,所述根据所述名称信息、所述参数信息和所述可执行文件生成模板,生成所述动态链接库文件,包括:根据所述名称信息,确定算子的运行场景;根据所述算子的运行场景,从所述可执行文件生成模板中确定与所述运行场景对应的目标模板;以及通过调用目标模板,根据所述名称信息和所述参数信息,生成动态链接库文件。3.根据权利要求2所述的方法,其中,所述通过调用目标模板,根据所述名称信息和所述参数信息,生成动态链接库文件,包括:通过调用目标模板,根据所述名称信息和所述参数信息,生成与所述算子的运行场景对应的源代码文件;以及根据所述算子在不同运行场景的多个源代码文件,生成所述动态链接库文件。4.根据权利要求3所述的方法,其中,所述根据所述算子在不同运行场景的多个源代码文件,生成动态链接库文件,包括:根据所述多个源代码文件的运行场景,对所述多个源代码文件进行分类,得到第一运行模式的源代码文件和第二运行模式的源代码文件;对所述第一运行模式的源代码文件进行编译,得到所述第一运行模式的动态链接库文件;以及对所述第二运行模式的源代码文件进行编译,得到所述第二运行模式的动态链接库文件。5.一种深度学习框架算子的调用方法,包括:根据用于调用目标算子的源代码,确定目标运行场景和目标算子,其中,所述目标运行场景指示了所述目标算子的调用方式和所述目标算子的运行模式;根据所述目标运行场景和所述目标算子,从动态链接库文件中得到目标可执行文件,其中,所述动态链接库文件是利用权利要求1
‑
4任一项所述的方法生成的;以及通过运行所述目标可执行文件,以执行所述目标算子的运算逻辑。6.根据权利要求1所述的方法,其中,所述根据用于调用目标算子的源代码,确定目标运行场景和目标算子,包括:根据所述源代码的算子名称字段,确定所述目标算子;以及根据所述源代码的输入模式和场景标识字段,确定所述目标运行场景。7.根据权利要求6所述的方法,其中,所述根据所述源代码的输入模式和场景标识字段,确定所述目标运行场景,包括:根据所述源代码的输入模式,确定所述目标算子的运行模式;
通过运行所述源代码,得到所述场景标识字段的识别结果;以及根据所述识别结果,确定所述目标算子的调用方式。8.根据权利要求1所述的方法,其中,所述动态链接库文件还包括算子、算子的核函数名称与运行场景的映射关系,所述根据所述目标运行场景和所述目标算子,从所述动态链接库文件中得到目标可执行文件,包括:根据所述目标算子、所述目标运行场景和所述映射关系,得到所述目标算子在所述目标运行场景的核函数名称;以及根据所述核函数名称,从所述动态链接库文件中得到目标可执行文件。9.一种动态链接库文件的生成装置,包括:获取模块,用于获取可执行文件生成模板;第一获得模块,用于从张量配置信息中得到算子核函数的名称信息;第二获得模块,用于根据所述算子核函数的名称信息,从算子配置信息中得到算子核函数的参数信息;以及生成模块,用于根据所述名称信息、所述参数信息和所述可执行文件生成模板,生成所述动态链接库...
【专利技术属性】
技术研发人员:贾宏宇,杨嘉彬,陈威行,张云飞,
申请(专利权)人:北京百度网讯科技有限公司,
类型:发明
国别省市:
还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。