昇腾社区首页
中文
注册
开发者
下载

接口简介

本节介绍mstx打点接口。可以自定义采集时间段或者关键函数的开始和结束时间点,识别关键函数或迭代等信息,对性能和算子问题快速定界。

默认情况下mstx API无任何功能,需要在用户应用程序中调用mstx API后,根据不同场景使能mstx打点功能,例如使用msprof命令行采集时配置--msproftx=on、使用AscendCL API采集时配置ACL_PROF_MSPROFTX以及Ascend PyTorch Profiler接口采集时配置mstx=True等。

  • 库文件libms_tools_ext.so路径:${INSTALL_DIR}/lib64/。
  • 使用头文件编译时,用户程序编译时需链接dl库。头文件ms_tools_ext.h路径:${INSTALL_DIR}/include/mstx。

${INSTALL_DIR}请替换为CANN软件安装后文件存储路径。若安装的Ascend-cann-toolkit软件包,以root安装举例,则安装后文件存储路径为:/usr/local/Ascend/ascend-toolkit/latest。

接口列表

表1 MindStudio mstx接口列表

接口名称

功能简介

mstxGetToolId

用于获取当前劫持mstx接口的工具ID。

mstxMarkA

标识瞬时事件。

mstxRangeStartA

标识时间段事件的开始。

mstxRangeEnd

标识时间段事件的结束。

mstxDomainCreateA

创建自定义domain。

mstxDomainDestroy

销毁指定的domain,销毁后的domain不能再次使用,需要重新创建。

mstxDomainMarkA

在指定的domain内,标记瞬时事件。

mstxDomainRangeStartA

在指定的domain内,标识时间段事件的开始。

mstxDomainRangeEnd

在指定的domain内,标识时间段事件的结束。

mstxMemHeapRegister

注册内存池。

mstxMemRegionsRegister

注册内存池二次分配。

mstxMemRegionsUnregister

注销内存池二次分配。

mstxMemHeapUnregister

注销内存池时,与之关联的Regions将一并被注销。