TRACE_START

产品支持情况

产品

是否支持

Ascend 950PR/Ascend 950DT

√

Atlas A3 训练系列产品/Atlas A3 推理系列产品

√

Atlas A2 训练系列产品/Atlas A2 推理系列产品

√

Atlas 200I/500 A2 推理产品

√

Atlas 推理系列产品AI Core

√

Atlas 推理系列产品Vector Core

x

Atlas 训练系列产品

√

功能说明

通过CAModel进行算子性能仿真时,可对算子任意运行阶段打点,从而分析不同指令的流水图,以便进一步性能调优。

用于表示起始位置打点,一般与TRACE_STOP配套使用。

该功能主要用于调试和性能分析,开启后会对算子性能产生一定影响,通常在调测阶段使用,生产环境建议关闭。

默认情况下,该功能关闭,开发者可以按需通过如下方式开启打点功能。

修改Kernel直调工程cmake目录下的npu_lib.cmake文件,在ascendc_compile_definitions命令中增加-DASCENDC_TRACE_ON编译选项,来开启打点功能。示例如下:
1
2
3
4
// 打开算子的打点功能
ascendc_compile_definitions(ascendc_kernels_${RUN_MODE} PRIVATE
    -DASCENDC_TRACE_ON
)

函数原型

1
2
#define TRACE_START(TraceId apid)
#define TRACE_START(pipe_t pipe, TraceId apid) // 该接口仅支持Ascend 950PR/Ascend 950DT

参数说明

参数名

输入/输出

描述

apid

输入

当前预留了十个用户自定义的类型:

  • 0x0:USER_DEFINE_0
  • 0x1:USER_DEFINE_1
  • 0x2:USER_DEFINE_2
  • 0x3:USER_DEFINE_3
  • 0x4:USER_DEFINE_4
  • 0x5:USER_DEFINE_5
  • 0x6:USER_DEFINE_6
  • 0x7:USER_DEFINE_7
  • 0x8:USER_DEFINE_8
  • 0x9:USER_DEFINE_9

pipe

输入

指定打点所在的pipeline类型

返回值说明

无

约束说明

调用示例

在Kernel代码中特定指令位置打上TRACE_START/TRACE_STOP:

1
2
3
TRACE_START(0x2);
Add(zLocal, xLocal, yLocal, dataSize);
TRACE_STOP(0x2);