msProbe 开发指南
本文面向 msProbe 的开发和维护人员,介绍源码目录、构建方式、功能开发流程、功能改动后的验证方法,以及资料联动更新要求。本文重点结合 msProbe 当前仓库和现有文档内容编写,适用于新增命令参数、扩展工具能力、增加交付件或维护软件包安装方式等场景。
1. msProbe 开发概述
msProbe 提供 AI 任务运行精度数据的采集、预检与比对等能力。围绕开发工作,通常可以分为以下几类:
开发对象 |
典型内容 |
|---|---|
采集能力 |
包括训练和推理模块的精度数据采集能力 |
预检能力 |
|
比对能力 |
|
溢出检测 |
|
扩展功能 |
训练前配置检查、训练状态监测、 checkpoint比对、趋势可视化等 |
文档资料 |
安装指南、快速入门、功能说明、数据文件参考、扩展功能 |
2. 代码目录
根据当前仓库资料,msProbe 项目主要目录如下:
目录 |
说明 |
|---|---|
|
C/C++源码目录 |
|
存放解析C化部分cmake文件 |
|
文档目录 |
|
工具配置样例存放目录 |
|
交付件生成目录 |
|
插件类代码总入口 |
|
工具核心功能模块 |
|
推理工具模块 |
|
MindSpore工具模块 |
|
msaccucmp工具模块 |
|
溢出检测模块 |
|
PyTorch工具模块 |
|
可视化模块 |
|
存放安装卸载升级脚本 |
|
测试代码目录 |
|
中文文档 |
3. 开发环境配置
按照《msProbe 安装指南 — 源码安装》章节完成编译和测试环境的搭建。
说明: 环境镜像的构建方法及配套软件版本由 MindStudio 统一镜像制作指南维护,本仓库不重复定义。
进入编译容器后,可执行启动界面提示的命令(如 py310)切换至对应的 Python 版本环境。
4. 获取代码与构建
4.1 获取代码
git clone https://gitcode.com/Ascend/msprobe.git
cd msprobe
4.2 编译安装基础工具包
python3 build.py
cd ./artifacts
pip install ./mindstudio_probe*.whl
编译工具包时还可以选择编译的功能模块,通过–include-mod参数配置,详见《msProbe工具安装指南》
安装完成后,建议立即校验:
which msprobe
msprobe --help
5. 测试与验证
仓库提供了统一的单元测试入口:
# Python单元测试
python3 build.py test
测试数据应该放在
test/目录下的相应位置。运行测试后,代码覆盖率报告生成在./report目录下。
6. 文档联动更新
功能开发完成后,若改动影响用户使用方式或输出结果,需要同步更新文档。
改动类型 |
需同步更新的文档 |
|---|---|
安装、编译、升级方式 |
|
快速入门 |
|
功能文档 |
|
性能基线文档 |
|
案例文档 |
|
FAQ |
|
若新增文档、截图或示意图:
图片统一放在
docs/zh/figures。文件名应与功能语义对应。
正文中的图标题、路径、说明文字要同步更新。
7. 提交流程建议
在功能开发完成后,先执行本地安装验证。
至少完成一轮
UT,必要时补充ST。若涉及用户可见行为变化,同步补充文档和示例命令。
若新增分析能力,说明其输入数据要求、输出文件和适用场景。