Skip to content

Ascend/msmodeling

Repository files navigation

MindStudio Modeling

昇腾 AI 模型性能建模与仿真工具

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

✨ 最新消息

持续支持 DeepSeek、Kimi、Qwen、GLM、MiniMax 等国内主流大模型,点击系列名称展开查看新增支持记录。

DeepSeek 系列

🔹 [2026.06.04] msModeling 新增 DeepSeek-V4 模型支持
🔹 [2026.04.20] msModeling 新增 DeepSeek V3.2 模型支持
🔹 [2025.09.06] msModeling 新增 DeepSeek V3 模型支持

Kimi 系列

🔹 [2026.06.11] msModeling 新增 Kimi-K2.6 模型支持
🔹 [2026.05.27] msModeling 新增 Kimi-K2.5 模型支持
🔹 [2025.09.06] msModeling 新增 Kimi-K2 模型支持

Qwen 系列

🔹 [2026.04.20] msModeling 新增 Qwen3.5 图片输入支持
🔹 [2026.03.31] msModeling 新增 Qwen3.5 Dense / MoE 文本输入支持
🔹 [2025.12.25] msModeling 新增 Qwen3 MoE 模型支持
🔹 [2025.09.18] msModeling 新增 Qwen3-Next 模型支持
🔹 [2025.08.18] msModeling 新增 Qwen3 Dense 模型支持

GLM 系列

🔹 [2026.06.04] msModeling 新增 GLM5.1 模型支持
🔹 [2026.04.30] msModeling 新增 GLM5 模型支持
🔹 [2026.03.31] msModeling 新增 GLM-4 MoE 模型支持

MiniMax 系列

🔹 [2025.12.18] msModeling 新增 MiniMax M2 模型支持

点击模块名称展开查看已支持特性

模型推理性能仿真

🔹 多硬件仿真(Atlas 800 A2/A3、Atlas 350 等昇腾设备),支持自定义设备画像
🔹 LLM Prefill / Decode 分阶段仿真
🔹 Prefix Cache 仿真
🔹 MTP 投机解码仿真
🔹 编译与图优化、多流通算掩盖
🔹 量化仿真(W8A8 / W4A8 / FP8 / MXFP4 等)
🔹 并行与 MoE 扩展(TP / DP / EP,及 Embedding TP、Vision TP 等细粒度并行)
🔹 性能模型切换(Roofline / Profiling)
🔹 Chrome Trace / Debug
🔹 视频生成 DiT 仿真(Ulysses、CFG、DiT Cache)

服务化性能仿真

🔹 LLM / VLM 约束下吞吐优化(TTFT / TPOT / 服务成本)
🔹 PD 模式(混部 / 分离 / 配比)
🔹 并行策略搜索(TP / EP / MOE-DP)
🔹 MTP 配置搜索
🔹 Chunked Prefill 仿真
🔹 Prefix Cache 仿真
🔹 变长负载仿真
🔹 多流通算掩盖
🔹 跨硬件对比

Web UI

🔹 LLM / VL 前向仿真与视频生成仿真
🔹 吞吐寻优实验(PD 混部 / 分离 / 配比)
🔹 命令预览与任务缓存
🔹 结果展示与导出(曲线、表格、显存/算子明细、Excel)

服务化实测寻优

🔹 服务框架实测寻优(PSO + Early Rejection)
🔹 多引擎支持(vLLM、MindIE)与评测策略
🔹 自定义寻优配置与断点续跑

ℹ️ 简介

MindStudio Modeling(msModeling)是专为昇腾 AI 处理器打造的神经网络推理性能仿真与分析框架,提供单模型性能仿真、服务级吞吐优化、服务化参数自动寻优与可视化分析能力,帮助开发者在无物理硬件或部署前期预测模型性能、识别瓶颈并优化配置。

⚙️ 功能介绍

msModeling 提供模型推理性能仿真、服务化性能仿真和服务化实测寻优等功能模块,覆盖相应性能仿真与寻优场景。模型与特性覆盖范围请参见《模型支持与特性支持矩阵》。

功能名称 功能描述
模型推理性能仿真 模型仿真模块,拦截 PyTorch 计算图,在指定设备画像上模拟推理过程,输出算子级性能分解、内存占用、算子 shape 及 Chrome Trace。
服务化性能仿真 吞吐优化仿真模块,在 SLO 约束下自动搜索最优并行策略与 batch 配置,支持 PD 混部、PD 分离、PD 配比三种模式。
服务化实测寻优 服务化实测寻优,基于 PSO 粒子寻优算法在 vLLM、MindIE 等真实服务框架上自动搜索满足时延约束的最优部署参数。

🚀 快速入门

以模型推理性能仿真与服务化性能仿真为例,快速跑通核心流程,请参见《模型推理性能仿真与服务化性能仿真快速入门》;也可通过《Web UI 使用指南》进行可视化交互配置与结果查看。

📦 安装指南

介绍工具的环境依赖与安装方法,请参见《msModeling 安装指南》。

📘 使用指南

各工具的详细使用说明请参阅其源码仓库中的 README 文件,也可通过上方功能介绍表格中的链接直接跳转。

💡 典型案例

通过典型问题场景帮助用户理解并掌握工具使用,请参见《模型推理性能仿真》与《服务化性能仿真》中的示例。

❓ FAQ

常见问题及解决方案,请提交 Issues 或参见各模块使用指南。

🌌 智能检索

为提升文档查阅效率,我们提供多种高效检索方式:
🔹 AI 问答(DeepWiki):自然语言问答,快速把握项目架构与模块关系。
🔹 AI 问答(ZRead):中文问答体验更优,精准定位功能用法与细节。
🔹 精确搜索(ReadTheDocs):关键词全文检索,直达接口、参数与报错等信息。

🛠️ 贡献指南

欢迎参与项目贡献!详细的贡献流程、代码规范、Commit 规范、测试要求等,请参见《CONTRIBUTING.md》。如有疑问,请提交 Issues

⚖️ 相关说明

🔹 《版本说明
🔹 《许可证声明
🔹 《安全声明
🔹 免责声明:本工具仿真与优化结果仅供性能评估参考,最终性能表现请以真实环境实测为准

🤝 建议与交流

欢迎大家为社区做贡献。如果有任何疑问或建议,请提交 Issues,我们会尽快回复。感谢您的支持。

SIG 例会:MindStudio Modeling Weekly Meeting 每周三 10:00-12:00(UTC+8)举行,会议纪要与议题请参见 sig-msit-modeling,也可使用 时区转换 查看本地时间。

即时互动(微信群) 官方资讯(公众号) 深度支持(助手/论坛)

扫码加入技术交流群

扫码关注官方公众号
扫码入群并关注公众号,直达 MindStudio 用户与开发者最快捷的交流平台:
快速提问: 与社区小伙伴即时探讨技术问题
掌握动态: 第一时间获取版本发布与功能更新通知
经验共享: 与广大开发者交流最佳实践与实战心得

更多支持渠道:👉 昇腾助手:WeChat 👉 昇腾论坛:Website

🙏 致谢

本工具由以下公司/部门联合贡献,以下排名不分先后:

🔹 华为
  昇腾计算产品部
  湛卢、AI Workload
  2012 网络技术实验室、2012 马尔科夫实验室
  小巧灵突击队、OTT 系统部

🔹 蚂蚁集团

🔹 电信研究院

感谢来自社区的每一个 PR,欢迎贡献!

About

No description, website, or topics provided.

Resources

Contributing

Stars

0 stars

Watchers

0 watching

Forks

Packages

 
 
 

Contributors

Languages