Profiling性能数据采集
功能介绍
用户可以在图加载和图执行过程中,采集Profiling性能数据,用于性能分析。使用时有三种方式:
序号 |
采集方式 |
|---|---|
方式一(不推荐使用) |
通过配置如下环境变量,采集Profiling性能数据:
|
方式二 |
通过GEInitialize传入option参数:
|
方式三 |
调用如下接口,采集Profiling性能数据:
如果需要采集迭代轨迹数据,还需要通过GEInitialize传入option参数ge.exec.profilingOptions或通过环境变量PROFILING_OPTIONS传入。传入字段包括training_trace/bp_point/fp_point。 |
支持的芯片型号
昇腾310P AI处理器
昇腾910 AI处理器
采集前配置
进行Profiling采集前,需要从完整样例参考中获取构建Graph并直接编译并运行Graph样例后,进行如下操作:
- 在源码文件“main.cpp”开头添加“#include "ge/ge_prof.h"”代码。
- 在编译脚本“Makefile”内的“LIBS”行下添加“-lmsprofiler”字段或在“CMakeLists.txt”内的“target_link_libraries”行下添加“msprofiler”字段。
定义接口的头文件 |
用途 |
对应的库文件 |
|---|---|---|
ge/ge_prof.h |
用于定义Profiling配置的接口。 |
libmsprofiler.so
说明:
ge头文件在“CANN软件安装后文件存储路径/include/”目录下,ge库文件在“CANN软件安装后文件存储路径/lib64/”目录下。 |
通过方式三采集性能数据
该特性为全局特性,不是session级特性,使能后在所有session均配置生效。
建议的接口调用顺序为:

调用示例为:
// 构造Graph,该步骤省略
// ......
// init ge
std::map<std::string, std::string> ge_options = {{"ge.socVersion", "xxx"}, {"ge.graphRunMode", "1"}};
ge::GEInitialize(ge_options);
std::string profilerResultPath = "/home/test/prof"; //该路径需要提前创建
uint32_t length = strlen("/home/test/prof");
ret = ge::aclgrphProfInit(profilerResultPath.c_str(), length);
std::map<string, string> options = {{"a", "b"}, {"c", "d"}};
uint32_t graphId = 0;
ge::Session *session = new Session(options);
ret = session->AddGraph(graphId, graph);
uint32_t deviceid_list[1] = {0};
uint32_t device_nums = 1;
uint64_t data_type_config = ProfDataTypeConfig::kProfTaskTime | ProfDataTypeConfig::kProfAiCoreMetrics | ProfDataTypeConfig::kProfAicpu | ProfDataTypeConfig::kProfTrainingTrace | ProfDataTypeConfig::kProfHccl | ProfDataTypeConfig:: kProfL2cache;
ProfAicoreEvents *aicore_events = NULL;
ProfilingAicoreMetrics aicore_metrics = ProfilingAicoreMetrics::kAicoreArithmeticUtilization;
ge::aclgrphProfConfig *pro_config = ge::aclgrphProfCreateConfig(deviceid_list, device_nums, aicore_metrics, aicore_events, data_type_config);
ge::aclgrphProfStart(pro_config);
session->RunGraph(graphId, inputs_r, outputs_r);
ge::aclgrphProfStop(pro_config);
ge::aclgrphProfDestroyConfig(pro_config);
ge::aclgrphProfFinalize();
delete session;
ge::GEFinalize();
父主题: 专题
