离线推理场景性能分析快速入门
离线推理场景下,推荐使用msprof命令行方式采集和解析性能数据,并通过生成的结果文件分析性能瓶颈。
前提条件
下文根据昇腾AI处理器的PCle工作模型区分为Ascend EP和Ascend RC两种操作步骤,Ascend EP和Ascend RC详细介绍请参见《昇腾产品形态说明》。
采集、解析并导出性能数据(Ascend EP)
- 登录装有Ascend-cann-toolkit开发套件包的运行环境,执行如下命令,可一键式采集、解析并导出性能数据:
msprof --output={path} {用户程序}
命令示例:
msprof --output=/home/HwHiAiUser/profiling_output /home/HwHiAiUser/HIAI_PROJECTS/MyAppname/out/main
表1 参数说明 参数
描述
可选/必选
--output
收集到的Profiling数据的存放路径,默认为AI任务文件所在目录。
路径中不能包含特殊字符:"\n", "\\n", "\f", "\\f", "\r", "\\r", "\b", "\\b", "\t", "\\t", "\v", "\\v", "\u007F", "\\u007F", "\"", "\\\"", "'", "\'", "\\", "\\\\", "%", "\\%", ">", "\\>", "<", "\\<", "|", "\\|", "&", "\\&", "$", "\\$", ";", "\\;", "`", "\\`"。
可选
传入用户程序
<app> [app arguments]
请根据实际情况在msprof命令末尾添加AI任务执行命令来传入用户程序或执行脚本。
格式:msprof [msprof arguments] <app> [app arguments]
- 举例1:msprof --output=/home/projects/output main
- 举例2:msprof --output=/home/projects/output /home/projects/MyApp/out/main
- 举例3:msprof --output=/home/projects/output /home/projects/MyApp/out/main parameter1 parameter2
- 举例4:msprof --output=/home/projects/output /home/projects/MyApp/out/sample_run.sh parameter1 parameter2
- 举例5:msprof --output=/home/projects/output python3 /home/projects/MyApp/out/sample_run.py parameter1 parameter2
必选
命令执行完成后,在--output指定的目录下生成PROF_XXX目录,存放自动解析后的性能数据(以下仅展示性能数据)。
1 2 3 4 5 6 7 8 9 10
├── host //保存原始数据,用户无需关注 │ └── data ├── device_{id} //保存原始数据,用户无需关注 │ └── data └── mindstudio_profiler_output ├── msprof_{timestamp}.json ├── step_trace_{timestamp}.json ├── xx_*.csv ... └── README.txt
- 进入mindstudio_profiler_output目录,查看性能数据文件。
默认情况下采集到的文件如表2所示。
表2 msprof默认配置采集的性能数据文件 文件名
说明
msprof_*.json
timeline数据总表。
step_trace_*.json
迭代轨迹数据,每轮迭代的耗时。单算子场景下无此性能数据文件。
op_summary_*.csv
AI Core和AI CPU算子数据。
op_statistic _*.csv
AI Core和AI CPU算子调用次数及耗时统计。
step_trace_*.csv
迭代轨迹数据。单算子场景下无此性能数据文件。
task_time_*.csv
Task Scheduler任务调度信息。
fusion_op_*.csv
模型中算子融合前后信息。单算子场景下无此性能数据文件。
api_statistic_*.csv
用于统计CANN层的API执行耗时信息。
注:“*”表示时间戳。
采集、解析并导出性能数据(Ascend RC)
- 登录运行环境,进入msprof工具所在目录“/var”。
- 执行如下命令采集性能数据。
msprof --output={path} {用户程序}
命令示例:
msprof --output=$[HOME]/profiling_output $[HOME]/HIAI_PROJECTS/MyAppname/out/main
表3 参数说明 参数
描述
可选/必选
--output
收集到的Profiling数据的存放路径,默认为AI任务文件所在目录。
路径中不能包含特殊字符:"\n", "\\n", "\f", "\\f", "\r", "\\r", "\b", "\\b", "\t", "\\t", "\v", "\\v", "\u007F", "\\u007F", "\"", "\\\"", "'", "\'", "\\", "\\\\", "%", "\\%", ">", "\\>", "<", "\\<", "|", "\\|", "&", "\\&", "$", "\\$", ";", "\\;", "`", "\\`"。
可选
传入用户程序
<app> [app arguments]
请根据实际情况在msprof命令末尾添加AI任务执行命令来传入用户程序或执行脚本。
格式:msprof [msprof arguments] <app> [app arguments]
- 举例1:msprof --output=/home/projects/output main
- 举例2:msprof --output=/home/projects/output /home/projects/MyApp/out/main
- 举例3:msprof --output=/home/projects/output /home/projects/MyApp/out/main parameter1 parameter2
- 举例4:msprof --output=/home/projects/output /home/projects/MyApp/out/sample_run.sh parameter1 parameter2
- 举例5:msprof --output=/home/projects/output python3 /home/projects/MyApp/out/sample_run.py parameter1 parameter2
必选
命令执行完成后,在--output指定的目录下生成PROF_XXX目录,目录结构如下。
1 2 3 4 5 6
├── device_{id} │ ├── data │ └── ... └── host │ ├── data │ └── ...
- 将PROF_XXX目录上传到安装toolkit包的开发环境,执行以下命令进行数据解析。
msprof --export=on --output=<dir>
参数
描述
可选/必选
--export
解析并导出性能数据。可选on或off,默认值为off。
若需导出个别模型(Model ID)/迭代(Iteration ID)的数据,可在msprof采集命令执行完成后重新执行msprof --export命令配置--model-id、--iteration-id参数。
对于未解析的PROF_XXX文件,自动解析后再导出。
示例:msprof --export=on --output=/home/HwHiAiUser
必选
--output
性能数据文件目录。须指定为PROF_XXX目录或PROF_XXX目录的父目录,例如:/home/HwHiAiUser/profiler_data/PROF_XXX。
路径中不能包含特殊字符:"\n", "\\n", "\f", "\\f", "\r", "\\r", "\b", "\\b", "\t", "\\t", "\v", "\\v", "\u007F", "\\u007F", "\"", "\\\"", "'", "\'", "\\", "\\\\", "%", "\\%", ">", "\\>", "<", "\\<", "|", "\\|", "&", "\\&", "$", "\\$", ";", "\\;", "`", "\\`"。
必选
PROF_XXX目录下会新增数据文件,目录结构如下:1 2 3 4 5 6 7 8 9
├── device_{id} │ ├── data ├── host │ ├── data │ └── ... └── mindstudio_profiler_output ├── xx_*.csv ├── xx_*.json ...
- 进入mindstudio_profiler_output目录,查看性能数据文件。
默认情况下采集到的文件如表4所示。
表4 msprof默认配置采集的性能数据文件 文件名
说明
msprof_*.json
timeline数据总表。
step_trace_*.json
迭代轨迹数据,每轮迭代的耗时。单算子场景下无此性能数据文件。
op_summary_*.csv
AI Core和AI CPU算子数据。
op_statistic _*.csv
AI Core和AI CPU算子调用次数及耗时统计。
step_trace_*.csv
迭代轨迹数据。单算子场景下无此性能数据文件。
task_time_*.csv
Task Scheduler任务调度信息。
fusion_op_*.csv
模型中算子融合前后信息。单算子场景下无此性能数据文件。
api_statistic_*.csv
用于统计CANN层的API执行耗时信息。
注:“*”表示时间戳。
性能分析
从上文我们可以看到,性能数据文件较多,分析方法也较灵活,以下介绍几个重要文件及分析方法。
- 通过msprof*.json文件从整体角度查看AI任务运行的时序信息,进而分析出可能存在的瓶颈点。图5 msprof*.json文件示例
- 区域1:CANN层数据,主要包含AscendCL、Runtime等组件以及Node(算子)的耗时数据。
- 区域2:底层NPU数据,主要包含Ascend Hardware下各个Stream任务流的耗时数据和迭代轨迹数据、昇腾AI处理器系统数据等。
- 区域3:展示timeline中各算子、接口的详细信息(单击各个timeline色块展示)。
从上图可以大致分析出耗时较长的API、算子、任务流等,并且根据对应的箭头指向找出对应的下发关系,分析执行推理过程中下层具体耗时较长的任务,查看区域3的耗时较长的接口和算子,再结合csv文件进行量化分析,定位出具体的性能瓶颈。
- 通过op_statistic_*.csv文件分析各类算子的调用总时间、总次数等,排查某类算子总耗时是否较长,进而分析这类算子是否有优化空间。图6 op_statistic_*.csv文件示例
可以按照Total Time排序,找出哪类算子耗时较长。
- 通过op_summary_*.csv文件分析具体某个算子的信息和耗时情况,从而找出高耗时算子,进而分析该算子是否有优化空间。图7 op_summary_*.csv文件示例
Task Duration字段为算子耗时信息,可以按照Task Duration排序,找出高耗时算子;也可以按照Task Type排序,查看不同核(AI Core和AI CPU)上运行的高耗时算子。