Skip to content

Ascend/msmemscope

Repository files navigation

MindStudio MemScope

昇腾 AI 内存调试调优利器

快速入门 AI问答(DeepWiki) AI问答(ZRead) 精确搜索 昇腾社区 报告问题

✨ 最新消息

🔹 [2026.04.29]:MindStudio Memscope 26.0.0版本上线!支持OOM场景采集显存快照。

🔹 [2026.04.08]:MindStudio Memscope 26.0.beta.1版本上线!显存占用一键拆解功能新增支持vllm、mindspeed场景。

🔹 [2026.02.01]:MindStudio Memscope 26.0.0-alpha.1版本上线! 支持Python API采集方式使用、支持PyTorch框架下采集内存快照、支持识别显存页表属性并进行落盘、支持获取Driver新增的显存分配接口。

🔹 [2025.12.30]:MindStudio Memscope 项目首次上线

ℹ️ 简介

MindStudio MemScope(内存分析工具,msMemScope)是基于昇腾硬件开发,用于模型训练与推理过程中的内存问题定位的工具。该工具提供内存泄漏检测、内存对比、内存块监测、内存拆解和低效内存识别等功能,帮助用户完成问题定位与处理。

⚙️ 功能介绍

msMemScope 工具提供内存采集、内存分析两大功能。

功能 功能说明 详细功能 使用场景及说明
内存采集 msMemScope 工具提供内存事件的采集能力,允许自定义设置采集内存范围和采集项,为后续分析提供原始数据。 Python 接口方式采集 支持通过 Python 接口采集信息,提供自定义设置采集内存范围和采集项,采集内存事件和 Python Trace 事件能力,实现精准采集、高效分析。
命令行方式采集 支持通过命令行方式采集信息,提供在非 Python 场景下采集内存事件与内存分析能力。
内存分析 msMemScope 工具基于采集的内存数据,提供泄漏、对比、监测、拆解、低效内存识别五项分析能力,帮助开发者快速诊断和优化内存问题。 内存泄漏分析 针对内存长时间未释放和内存泄漏等问题,需要进行内存分析时,msMemScope 工具提供内存泄漏分析和 kernelLaunch 粒度的内存变化分析功能,进行告警定位与分析。
内存对比分析 当两个 Step 内存使用存在差异时,可能会导致内存使用过多,甚至出现 OOM(Out of Memory,内存溢出)的问题,则需要使用 msMemScope 工具的内存对比分析功能来定位并分析问题。
内存块监测 在大模型场景中,当遇到内存踩踏定位困难时,msMemScope 工具支持通过 Python 接口和命令行两种方式,在算子执行前后对指定的内存块进行监测。根据内存块数据的变化,快速确定算子间内存踩踏的范围或具体位置。
内存拆解 msMemScope 工具提供内存拆解功能,支持对 CANN 层和 Ascend Extension for PyTorch 框架的内存使用情况进行拆解,输出模型权重、激活值、梯度,以及优化器等组件的内存占用情况。
低效内存识别 在训练推理模型过程中,可能会存在部分内存块申请后未立即使用,或使用完毕后未及时释放的低效情况。msMemScope 工具可帮助识别这种低效内存的使用现象,从而优化训练推理模型。

🚀 快速入门

快速入门旨在帮助用户快速熟悉 msMemScope 工具的使用方式,请参见 《msMemScope 快速入门》。

📦 安装指南

msMemScope 工具支持通过软件包和源码两种方式进行安装,可根据实际需求选择合适的安装方式,请参见 《msMemScope 安装指南》。

msMemScope 工具当前支持 CANN、Ascend Extension for PyTorch、MindSpore 以及 Aten 算子的内存采集,具体版本支持情况如下表所示。

产品 说明
CANN CANN 8.2.RC1 及之后版本的 ATB 算子(Ascend Transformers Boost)。
Ascend Extension for PyTorch Ascend Extension for PyTorch 7.0.0 及之后版本。
MindSpore MindSpore 2.7.0 及之后版本。
Aten 算子 当采集 Aten 算子下发与访问事件时,需使用 PyTorch 2.3.1 或更高版本。

📘 使用指南

工具的详细使用方法,请参见 《msMemScope 使用指南》。

💡 典型案例

通过典型问题场景帮助用户理解并掌握工具使用,请参见 《msMemScope 典型案例》。

📚 API 参考

msMemScope 工具提供 API 接口,便于快速分析内存情况,请参见 《API 参考》。

🛠️ 贡献指南

欢迎参与项目贡献,请参见 《贡献指南》。

🌌 智能检索

为提升文档查阅效率,我们提供多种高效检索方式:

⚖️ 相关说明

🔹 《版本说明

发布版本 发布时间 发布 Tag 兼容性说明
26.0.0 2026/04/29 tag_MindStudio_26.0.0.B120_001 兼容昇腾 CANN 8.5.0 及以前版本。请参考《CANN 安装指南》获取 CANN 安装包。
26.0.beta.1 2026/04/08 tag_MindStudio_26.0.T2.B100_001 兼容昇腾 CANN 8.5.0 及以前版本。请参考《CANN 安装指南》获取 CANN 安装包。
26.0.0-alpha.1 2026/02/04 tag_MindStudio_26.0.0-alpha.1 兼容昇腾 CANN 8.5.0 及以前版本。请参考《CANN 安装指南》获取 CANN 安装包。

新增特性

  • 支持 PyTorch 框架下采集内存快照。
  • 支持识别显存页表属性并进行落盘。
  • 支持 vLLM、Verl、MindSpeed 场景下的显存拆解。
📖 查看历史版本

更多历史版本信息请参见 历史版本

🔹 《许可证声明》 🔹 《安全声明》 🔹 《免责声明

🤝 建议与交流

欢迎大家为社区做贡献。如果有任何疑问或建议,请提交 Issues,我们会尽快回复。也可参考《交流指南》获取详细的联系方式和支持渠道。感谢您的支持。

📱 关注 MindStudio 公众号 💬 更多交流与支持

扫码关注获取最新动态
💡 加入微信交流群
关注公众号,回复"交流群"即可获取入群二维码。

🛠️ 其他渠道
👉 昇腾助手:WeChat
👉 昇腾论坛:Website

🙏 致谢

本工具由华为 MindStudio 全流程开发工具链团队贡献,致力于提供端到端的昇腾 AI 应用开发解决方案,使能开发者高效完成训练开发、推理开发和算子开发。 感谢来自社区的每一个 PR,欢迎贡献 msMemScope。

About

No description, website, or topics provided.

Resources

License

Stars

0 stars

Watchers

0 watching

Forks

Packages

 
 
 

Contributors