Profiling数据采集功能

功能介绍

性能分析工具(Profiling)用于采集和分析算子在昇腾AI处理器上运行阶段的关键性能指标,用户可根据输出的性能数据,快速定位软、硬件性能瓶颈,提升算子运行性能。

该功能可以与上板时间戳打点功能配套使用,通过timeline展示算子在每个核上的运行起始/终止时间等信息,方便更精准分析算子性能耗时。

  • 该功能仅在NPU板端运行场景下开启,其他场景不支持。
  • 该功能暂不支持与PRINTF/DumpTensor/DumpAccChkPoint/assert打印功能同时开启。

使用方法(命令行)

通过命令行进行性能数据采集的关键步骤如下,详细样例参考上板Profiling数据采集

  1. 执行如下命令开启数据采集,这里仅提供关键配置项示例,其他参数请参考NPU调测参数按需配置。

    • 开启Profiling数据采集功能
      • 基础功能:无需配置,默认采集PipeUtilization数据。
        ascendebug kernel --backend npu --profiling ... {其他NPU调测参数}
      • 高级功能(可选):如需采集其他数据,可进一步设置${profiling_metrics}。
        ascendebug kernel --backend npu --profiling ${profiling_metrics} ... {其他NPU调测参数}
    • 开启Profiling数据采集和时间戳打点功能
      1. 在核函数代码中按需调用AscendCTimeStamp自定义打点标识,descId取值参见表2,样例如下:
         AscendC::AscendCTimeStamp(descId);
      2. NPU调测命令示例如下:
        • 基础功能:无需配置,默认采集PipeUtilization数据。
          ascendebug kernel --backend npu  --dump-mode time_stamp --profiling ... {其他NPU调测参数}
        • 高级功能(可选):如需采集其他数据,可进一步设置${profiling_metrics}。
          ascendebug kernel --backend npu  --dump-mode time_stamp --profiling ${profiling_metrics} ... {其他NPU调测参数}

    其中${profiling_metrics}支持如下取值:

    • Atlas 200/300/500 推理产品 :ArithmeticUtilization、PipeUtilization(默认值)、Memory、MemoryL0、MemoryUB、ResourceConflictRatio
    • Atlas 200I/500 A2 推理产品 :ArithmeticUtilization、PipeUtilization、Memory、MemoryL0、MemoryUB、ResourceConflictRatio、L2Cache、PipelineExecuteUtilization(默认值)
    • Atlas 推理系列产品 :ArithmeticUtilization、PipeUtilization(默认值)、Memory、MemoryL0、MemoryUB、ResourceConflictRatio
    • Atlas 训练系列产品 :ArithmeticUtilization、PipeUtilization(默认值)、Memory、MemoryL0、MemoryUB、ResourceConflictRatio
    • Atlas A2 训练系列产品/Atlas 800I A2 推理产品 :ArithmeticUtilization、PipeUtilization(默认值)、Memory、MemoryL0、MemoryUB、ResourceConflictRatio、L2Cache

    更多关于数据采集高级功能的介绍可参考性能调优工具指南的“采集AI任务运行性能数据”章节中--aic-metrics配置项。

  2. 检查是否成功使能Profiling。

    当打屏日志中有调用msprof开启profiling的调用信息,并且有“npu kernel run end” 信息则表示Profiling执行完成。

  3. 查看结果文件,详细说明参见产物说明

使用方法(API)

通过API开启性能数据采集的关键步骤如下,详细样例参考上板Profiling数据采集

  1. 完成环境搭建,并准备好输入/标杆数据文件。

    核函数直调工程场景下,需提前配置好核函数源码信息OpKernelInfo

  2. 构建算子信息。调用DebugOp类里input系列接口(如tensor_input、custom_input等),设置算子的输入、输出、属性值等信息。
  3. 创建调试对象并初始化工作空间。调用create_op_executor接口,创建调试对象op_executor,用户可传入${work_dir}参数手动设置工作空间。
  4. (可选)参考Tiling调测功能 > 使用方法(API)章节,调用Tiling调测接口生成Tiling Info文件。

    本步骤仅适用于没有Tiling Info文件的场景,需调用Tiling调测API生成Tiling Info。

  5. 调用NPU编译接口和Profiling运行接口。

    1. 通过CompileNpuOptions设置NPU编译相关的功能,如内存检测功能。
    2. 根据不同场景调用对应的NPU编译接口。
    3. 通过RunProfilingOptions设置性能数据采集相关的功能,并通过NpuCompileInfo设置NPU编译属性。
    4. 调用run_profiling接口。

    接口调用示例如下:

    • 开启Profiling数据采集功能
      # 调用NPU编译接口
      compile_npu_options = ascendebug.CompileNpuOptions()
      name, kernel_file, extern = op_executor.compile_custom_npu(customize_path, tiling_info.tiling_key, compile_npu_options)
      # 配置Profiling属性,调用Profiling运行接口
      profiling_options = ascendebug.RunProfilingOptions(block_num=24, loop=10)
      npu_compile_info = ascendebug.NpuCompileInfo(syncall=extern['cross_core_sync'], task_ration=extern['task_ration'])
      op_executor.run_profiling(kernel_file, profiling_options, npu_compile_info=npu_compile_info, tiling_info=tiling_info)
    • 开启Profiling数据采集和时间戳打点功能
      1. 在核函数代码中按需调用AscendCTimeStamp自定义打点标识,descId取值参见表2,样例如下:
         AscendC::AscendCTimeStamp(descId);
      2. 设置dump_model为“time_stamp”,样例如下:
        # 调用NPU编译接口
        compile_npu_options = ascendebug.CompileNpuOptions(dump_mode='time_stamp')
        name, kernel_file, extern = op_executor.compile_custom_npu(customize_path, tiling_info.tiling_key, compile_npu_options)
        # 配置Profiling属性,调用Profiling运行接口
        profiling_options = ascendebug.RunProfilingOptions(block_num=24, loop=10)
        npu_compile_info = ascendebug.NpuCompileInfo(syncall=extern['cross_core_sync'], task_ration=extern['task_ration'], dump_mode='time_stamp)
        op_executor.run_profiling(kernel_file, profiling_options, npu_compile_info=npu_compile_info, tiling_info=tiling_info)

  6. 查看结果文件,详细说明参见产物说明

产物说明

无论是命令行方式或API方式,Profiling解析结果存放在${root}/${work_dir}/npu/output路径下,其中${root}表示当前操作路径,${work_dir}表示调测工作空间,默认为/debug_workspace/${op_type}目录,${op_type}为算子名。目录结构示例如下,默认情况下采集到的文件如表1所示。

├ ${op_type}               // 算子名
├── npu
│   ├── build           // 存放NPU编译生成的中间文件
│       ├── launch_args.so
│   ├── output         // 存放NPU编译运行的输出文件及精度比对结果
│       ├── y.bin      // 运行输出原始数据
│       ├── y.txt      // 精度比对结果文件
│       ├── PROF_00000x_xxxx_xxx        // 存放Profiling结果文件
│           ├── mindstudio_profiler_output
│               ├── task_time_*.csv
│               ├── prof_rule_*.json
│               ├── op_summary_*.csv
│               ├── op_statistic_*.csv
│               ├── api_statistic_*.csv
│               ├── step_trace_*.csv
│               ├── step_trace_*.json
│               ├── msprof_*.json      // 若开启时间戳打点,会包含该信息       
│   ├── src          // 存放NPU编译生成的临时代码文件
│       ├── _gen_args_${op_type}.cpp
表1 Profiling性能数据文件说明

文件名

说明

备注

task_time_*.csv

Task Scheduler任务调度信息。

关于性能文件的详细介绍可参见性能调优工具指南中“性能数据文件参考”。

prof_rule_*.json

调优建议。

op_summary_*.csv

AI Core和AI CPU算子数据。

op_statistic _*.csv

AI Core和AI CPU算子调用次数及耗时统计。

api_statistic_*.csv

用于统计CANN层的API执行耗时信息。

step_trace_*.json

(可选)迭代轨迹数据,每轮迭代的耗时。单算子场景下无此性能数据文件。

step_trace_*.csv

(可选)迭代轨迹数据。单算子场景下无此性能数据文件。

msprof_*.json

timeline数据总表。

说明:

若开启时间戳打点功能,生成的${root}/${work_dir}/npu/dump/PARSER_*/dump_data/${core_num}/time_stamp_core_${core_num}.csv文件会被解析并添加到msprof_*.json文件末尾。

注:“*”表示{timestamp}时间戳。