【技术实现步骤摘要】
模型推理性能的分析方法、装置及电子设备
[0001]本公开涉及人工智能
,尤其涉及深度学习、模型推理等
,具体涉及一种模型推理性能的分析方法、装置及电子设备。
技术介绍
[0002]目前,机器学习与深度学习的相关技术在自然语言处理、计算机视觉等众多领域取得了巨大成功,也逐渐在各行各业中落地。性能剖析(Profiling)是指在程序的执行过程中,收集能反映程序执行状态的数据,例如程序执行所占用的内存、特定指令的使用情况或函数调用的频率和持续时间等,帮助定位和优化程序性能问题。
技术实现思路
[0003]本公开提供了一种模型推理性能的分析方法、装置及电子设备。
[0004]根据本公开的第一方面,提供了一种模型推理性能的分析方法,包括:
[0005]确定待分析模型;
[0006]接收第一输入,并获取所述第一输入中的目标信息,所述目标信息包括插桩代码的位置信息或针对预先嵌入的性能分析器的启动指令;
[0007]基于所述目标信息执行对所述待分析模型的推理性能分析,并生成分析结果文件。
[0008]根据本公开的第二方面,提供了一种模型推理性能的分析装置,包括:
[0009]确定模块,用于确定待分析模型;
[0010]获取模块,用于接收第一输入,并获取所述第一输入中的目标信息,所述目标信息包括插桩代码的位置信息或针对预先嵌入的性能分析器的启动指令;
[0011]分析模块,用于基于所述目标信息执行对所述待分析模型的推理性能分析,并生成分析结果 ...
【技术保护点】
【技术特征摘要】
1.一种模型推理性能的分析方法,包括:确定待分析模型;接收第一输入,并获取所述第一输入中的目标信息,所述目标信息包括插桩代码的位置信息或针对预先嵌入的性能分析器的启动指令;基于所述目标信息执行对所述待分析模型的推理性能分析,并生成分析结果文件。2.根据权利要求1所述的方法,其中,所述目标信息包括插桩代码的位置信息,所述位置信息包括所述插桩代码在所述待分析模型对应的模型代码中插桩的起始行号和终止行号;所述基于所述目标信息执行对所述待分析模型的推理性能分析,包括:确定插桩代码;基于所述起始行号和所述终止行号,在所述模型代码中的所述起始行号之前和所述终止行号之后分别插入所述插桩代码;在运行所述待分析模型的情况下,基于所述插桩代码,对所述模型代码中位于所述起始行号和所述终止行号之间的代码进行推理性能分析。3.根据权利要求2所述的方法,其中,在所述目标信息包括多个插桩代码各自对应的位置信息的情况下,所述在所述模型代码中的所述起始行号之前和所述终止行号之后分别插入插桩代码,包括:检测所述多个插桩代码各自对应的位置信息是否正确;在所述多个插桩代码各自对应的位置信息正确的情况下,在所述模型代码中的目标位置处分别插入对应的插桩代码,所述目标位置为所述多个插桩代码各自对应的所述起始行号和所述终止行号。4.根据权利要求3所述的方法,其中,所述检测所述多个插桩代码各自对应的位置信息是否正确,包括如下至少一项:针对每个插桩代码,检测该插桩代码对应的起始行号和终止行号是否位于所述模型代码对应的行号范围内,其中,在该插桩代码对应的起始行号和终止行号位于所述模型代码对应的行号范围内的情况下,该插桩代码对应的位置信息正确;针对每个插桩代码,检测该插桩代码对应的起始行号和终止行号所包括的代码范围是否与其他插桩代码对应的起始行号和终止行号所包括的代码范围出现重叠,其中,在未出现重叠的情况下,该插桩代码对应的位置信息正确。5.根据权利要求2所述的方法,其中,所述生成分析结果文件,包括:生成分析结果文件,并将所述分析结果文件中的代码行号转换为所述模型代码未插入所述插桩代码之前的行号。6.根据权利要求1所述的方法,其中,所述目标信息包括插桩代码的位置信息,所述位置信息基于所述待分析模型对应的模型代码中的函数名确定;所述基于所述目标信息执行对所述待分析模型的推理性能分析,包括:在所述模型代码中所述函数名对应的函数代码前和所述函数代码后分别插入插桩代码;在运行所述待分析模型的情况下,基于所述插桩代码,对所述模型代码中所述函数名对应的函数代码进行推理性能分析。
7.根据权利要求1所述的方法,其中,所述目标信息包括针对预先嵌入的性能分析器的启动指令;所述基于所述目标信息执行对所述待分析模型的推理性能分析,包括:确定预先写入的所述性能分析器对应的目标代码;在获取到针对所述性能分析器的启动指令的情况下,基于所述启动指令运行所述性能分析器对应的目标代码,以执行对所述待分析模型的推理性能分析。8.一种模型推理性能的分析装置,包括:确定模块,用于确定待分析模型;获取模块,用于接收第一输入,并获取所述第一输入中的目标信息,所述目标信息包括插桩代码的位置信息或针对预先嵌入的性能分析器的启动指令;分析模块,用于基于所述...
【专利技术属性】
技术研发人员:王宽,施恩,钱正宇,胡鸣人,袁正雄,李金麒,褚振方,黄悦,罗阳,王国彬,钱洋,
申请(专利权)人:北京百度网讯科技有限公司,
类型:发明
国别省市:
还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。