msPTI 快速入门

1. 概述

msPTI(MindStudio Profiler Tools Interface)是华为昇腾 MindStudio 提供的一套性能剖析 API 集合。您可以使用 msPTI 构建面向 NPU 应用的性能分析工具,应用于推理和训练场景。

建议按照 安装工具 → 配置环境 → 运行样例 的顺序完成快速体验。


2. 环境准备

2.1 硬件环境

2.2 软件环境

  • CANN(含 msPTI):建议优先安装 CANN 软件(Toolkit + ops 包),msPTI 已集成于 CANN 中。若已安装 CANN,则可直接使用。
  • Python 环境:若使用 msPTI Python API,请确保 Python 3.10+ 环境已配置。
  • PyTorch + torch_npu(可选):Python Monitor 样例依赖 PyTorch 框架和 Ascend Extension for PyTorch(安装指南)。

2.3 约束说明

msPTI 不可与其他性能数据采集工具同时使用,否则会导致采集的数据丢失。


3. 工具安装

msPTI 已集成于 CANN 中,若已安装 CANN 且无需升级此工具,可直接跳过本步骤。

如需单独升级或安装最新版本,请参见《msPTI 工具安装指南》。

安装完成后,执行以下命令验证:

pip show mspti

若输出版本信息且无报错,则表示安装成功。


4. 快速运行样例

4.1 配置 CANN 环境变量

source ${install_path}/set_env.sh

其中 ${install_path} 为 CANN 安装路径,例如 /usr/local/Ascend/cann

4.2 运行基础样例

进入 Activity API 基础样例目录并执行:

cd ${install_path}/tools/mspti/samples/mspti_activity
bash sample_run.sh

运行成功后输出示例如下:

...
========== UserBufferRequest ============
result[0] is: 1.200000
result[1] is: 2.200000
result[2] is: 3.200000
result[3] is: 5.400000
result[4] is: 6.400000
result[5] is: 7.400000
result[6] is: 9.600000
result[7] is: 10.600000
========== UserBufferComplete ============
[RUNTIME_API] name: DevMalloc, start: 1775186328012443375, end: 1775186328012485525, ...
[MEMORY] operationType: ALLOCATION, memoryKind: MEMORY_DEVICE, ...
[RUNTIME_API] name: MemCopySync, start: 1775186328012545645, end: 1775186328012596965, ...
[MEMCPY] copyKind: HTOD, bytes: 32, ...
...