PrintTimeStamp

函数功能

提供时间戳打点功能,用于在算子Kernel代码中标记关键执行点,调用后会打印如下信息:

打印示例如下:

1
descId is 65577, rsv is 0, timeStamp is 13806084506158, pcPtr is 20619064414544.

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

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

    修改cmake目录下的npu_lib.cmake文件,在ascendc_compile_definitions命令中增加-DASCENDC_TIME_STAMP_ON,打开时间戳打点功能,示例如下:

    1
    2
    3
    ascendc_compile_definitions(ascendc_kernels_${RUN_MODE} PRIVATE
        -DASCENDC_TIME_STAMP_ON
    )
    

    CPU域调试时,不支持该功能。

  • 自定义算子工程
    修改算子工程op_kernel目录下的CMakeLists.txt文件,首行增加编译选项-DASCENDC_TIME_STAMP_ON,打开时间戳打点功能,示例如下:
    1
    add_ops_compile_options(ALL OPTIONS -DASCENDC_TIME_STAMP_ON)
    

函数原型

1
void PrintTimeStamp(uint32_t descId)

参数说明

参数名

输入/输出

描述

descId

输入

用户自定义标识符(自定义数字),用于区分不同打点位置

注意:

[0, 0xffff]是预留给Ascend C内部各个模块使用的id值,用户自定义的descId建议使用大于0xffff的数值。

返回值

支持的型号

Atlas A2 训练系列产品/Atlas 800I A2 推理产品/A200I A2 Box 异构组件

约束说明

调用示例

1
AscendC::PrintTimeStamp(65577);

打印结果如下:

1
2
3
opType=AddCustom, DumpHead: AIV-0, CoreType=AIV, block dim=8, total_block_num=8, block_remain_len=1047136, block_initial_space=1048576, rsv=0, magic=5aa5bccd
...// 一些框架内部的打点信息
descId is 65577, rsv is 0, timeStamp is 13806084506158, pcPtr is 20619064414544.