一文理清MindStudio训练推理工具资料全景大纲!
收藏回复举报
一文理清MindStudio训练推理工具资料全景大纲!
发表于2024-09-12 15:37:34
0 查看
  •  训练工具全景旅程图 

  • 推理工具全景旅程图

  • 训练工具资料大纲
工具说明资料承载资料链接
分析迁移
自动迁移工具支持将PyTorch训练脚本从GPU平台迁移至昇腾NPU平台Gitee开源https://gitee.com/ascend/mstt/wikis/%E5%B7%A5%E5%85%B7%E4%BB%8B%E7%BB%8D/%E5%88%86%E6%9E%90%E8%BF%81%E7%A7%BB%E5%B7%A5%E5%85%B7/%E5%88%86%E6%9E%90%E8%BF%81%E7%A7%BB%E5%B7%A5%E5%85%B7%E4%BB%8B%E7%BB%8D
脚本迁移工具(PyTorch GPU2Ascend)支持将PyTorch训练脚本从GPU平台迁移至昇腾NPU平台Gitee开源
脚本分析工具(PyTorch Analyse)提供分析脚本,帮助用户在执行迁移操作前,分析基于GPU平台的PyTorch训练脚本中算子、三方库套件、亲和API分析以及动态shape的支持情况Gitee开源
训推一体权重转换工具支持在NPU和GPU上训练好的模型转成Huggingface格式bin文件,用于推理使用Gitee开源
分析迁移工具提供PyTorch/TensorFlow训练脚本一键式迁移至昇腾NPU或MindSpore生态的功能社区https://www.hiascend.com/document/detail/zh/mindstudio/70RC2/T&ITools/training&inferencetools/atlasfmkt_16_0001.html
精度分析
MindStudio精度调试工具(MindStudio Probe)MindStudio Training Tools工具链下精度调试部分的工具包,提供精度数据采集、预检以及溢出检测等功能,用于辅助用户快速定位模型精度问题Gitee开源https://gitee.com/ascend/mstt/tree/master/debug/accuracy_tools
Ascend模型精度预检工具(api_accuracy_checker)在昇腾NPU上扫描用户训练模型中所有API,输出精度情况的诊断和分析Gitee开源
PyTorch精度工具(ptdbg_ascend)能进行PyTorch整网API粒度的数据dump、精度比对和溢出检测,从而定位PyTorch训练场景下的精度问题Gitee开源
精度调试工具提供自有实现算子的运算结果与业界标杆算子运算结果之间差异比对的工具社区https://www.hiascend.com/document/detail/zh/mindstudio/70RC2/T&ITools/training&inferencetools/atlasaccuracy_16_0001.html
性能分析
性能比对工具(compare_tools)支持比较GPU与NPU之间、NPU与NPU之间的性能差异Gitee开源https://gitee.com/ascend/mstt/tree/master/profiler
集群分析工具(cluster_analyse)在集群场景下,通过此工具来进行集群数据的分析Gitee开源
专家建议工具(advisor)将Ascend PyTorch Profiler或者msprof采集的PyThon场景性能数据进行分析,并输出性能调优建议Gitee开源
可视化呈现(Tensorboard)PyTorch profiling数据以及可视化的TensorBoard的插件Gitee开源https://gitee.com/ascend/mstt/tree/master/plugins/tensorboard-plugins/tb_plugin
性能分析工具用于采集和分析运行在昇腾AI处理器上的AI任务各个运行阶段的关键性能指标社区https://www.hiascend.com/document/detail/zh/mindstudio/70RC2/T&ITools/training&inferencetools/atlasprofiling_16_0001.html
扩展功能
AMCT工具一个针对昇腾芯片亲和的深度学习模型压缩工具包社区https://www.hiascend.com/document/detail/zh/mindstudio/70RC2/T&ITools/training&inferencetools/auxiliarydevtool_0017.html
AOE工具一款自动调优工具,作用是充分利用有限的硬件资源,以满足算子和整网的性能要求社区https://www.hiascend.com/document/detail/zh/mindstudio/70RC2/T&ITools/training&inferencetools/auxiliarydevtool_0010.html
HCCL性能测试工具分布式训练场景下,开发者可以通过此工具测试HCCL集合通信的功能正确性以及性能社区https://www.hiascend.com/document/detail/zh/mindstudio/70RC2/T&ITools/training&inferencetools/auxiliarydevtool_0018.html
算子及模型速查工具用于查询当前版本CANN支持的算子信息和ModelZoo支持的模型信息社区https://www.hiascend.com/document/detail/zh/mindstudio/70RC2/T&ITools/training&inferencetools/atlasopmodelquery_16_0001.html

  • 推理工具资料大纲
一级目录说明资料承载资料链接
迁移分析
模型支持度分析工具(msit analyze)Analyze分析工具提供从Caffe、TensorFlow以及ONNX模型迁移至昇腾硬件平台上的模型支持度分析功能。涵盖ATC支持度分析、昇腾算子速查工具分析以及ONNXChecker算子约束分析三大分析手段。Gitee开源https://gitee.com/ascend/msit/tree/master/msit/docs/analyze
模型转换
LLM迁移工具(msit llm transform)支持 PyTorch transformers LLM 模型迁移生成 atb 浮点模型,或由 LLM 的浮点模型 layer 层代码,迁移生成稀疏量化 layer 层代码,并输出大模型迁移分析报告。Gitee开源https://gitee.com/ascend/msit/blob/master/msit/docs/llm/%E5%B7%A5%E5%85%B7-llm%E6%A8%A1%E5%9E%8B%E8%BF%81%E7%A7%BB%E5%88%86%E6%9E%90%E4%BD%BF%E7%94%A8%E8%AF%B4%E6%98%8E.md
ATC工具是异构计算架构CANN体系下的模型转换工具, 可将开源框架的网络模型以及Ascend IR定义的单算子描述文件(json格式)转换为昇腾AI处理器支持的.om格式离线模型。社区https://www.hiascend.com/document/detail/zh/mindstudio/70RC2/T&ITools/training&inferencetools/auxiliarydevtool_0007.html
模型压缩
模型量化工具(msModelSlim)支持训练加速和推理加速,包括模型低秩分解、稀疏训练、训练后量化、量化感知训练等功能,昇腾AI模型开发用户可以灵活调用Python API接口,对模型进行性能调优,并支持导出不同格式模型,在昇腾AI处理器上运行。Gitee开源https://gitee.com/ascend/msit/tree/master/msmodelslim
社区https://www.hiascend.com/document/detail/zh/mindstudio/70RC2/T&ITools/training&inferencetools/modelslim_0001.html
AMCT工具是一个针对昇腾芯片亲和的深度学习模型压缩工具包,提供量化、稀疏等多种模型压缩特性,压缩后模型体积变小,部署到昇腾AI处理器上后可使能低比特运算,提高计算效率。社区https://www.hiascend.com/document/detail/zh/mindstudio/70RC2/T&ITools/training&inferencetools/auxiliarydevtool_0017.html
自动调优
传统模型自动调优(msit debug surgeon)surgeon 通过内置知识库的自动调优,使能ONNX模型在昇腾芯片的优化,并提供基于ONNX的改图功能。Gitee开源https://gitee.com/ascend/msit/tree/master/msit/docs/debug/surgeon
AOE工具是一款自动调优工具,通过生成调优策略、编译、在运行环境上验证的闭环反馈机制,不断迭代出更优的调优策略,最终得到最佳的调优策略。社区https://www.hiascend.com/document/detail/zh/mindstudio/70RC2/T&ITools/training&inferencetools/auxiliarydevtool_0010.html
精度分析
Bad Case分析工具(msit llm bc-analyze)msIT 工具针对两个模型在同一数据集下,表现不一致的情况,提供脚本和命令行两种交互方式来进行自动 bad case 分析,使能用户快速定位。Gitee开源https://gitee.com/ascend/msit/blob/500032ef926505894a5158ceb8ad51f63dd0aa85/msit/docs/llm/%E5%B7%A5%E5%85%B7-%E5%A4%A7%E6%A8%A1%E5%9E%8B%E7%B2%BE%E5%BA%A6%E5%AE%9A%E4%BD%8D%E4%B9%8BBadCase%E5%88%86%E6%9E%90%E5%B7%A5%E5%85%B7.md
LLM数据dump工具(msit llm dump)支持dump加速库模型推理过程中产生的中间数据,包括模型拓扑信息、Tensor信息、kernel 信息等。Gitee开源https://gitee.com/ascend/msit/blob/master/msit/docs/llm/%E5%B7%A5%E5%85%B7-DUMP%E5%8A%A0%E9%80%9F%E5%BA%93%E6%95%B0%E6%8D%AE%E4%BD%BF%E7%94%A8%E8%AF%B4%E6%98%8E.md
支持dump pyTorch模型推理数据,包括tensor信息、模型拓扑信息等,提供python API来使能数据dump。https://gitee.com/ascend/msit/blob/master/msit/docs/llm/%E5%B7%A5%E5%85%B7-DUMP%E5%9C%A8%E7%BA%BF%E6%8E%A8%E7%90%86%E6%95%B0%E6%8D%AE%E4%BD%BF%E7%94%A8%E8%AF%B4%E6%98%8E.md
传统模型精度比对工具(msit debug compare)compare一键式全流程精度比对(推理)功能将推理场景的精度比对做了自动化,适用于 TensorFlow 和 ONNX 模型,用户只需要输入原始模型,对应的离线模型和输入,输出整网比对的结果。Gitee开源https://gitee.com/ascend/msit/tree/master/msit/docs/debug/compare
LLM精度比对工具(msit llm compare)提供模型推理数据与业界标杆数据比对的能力,适用于LLM模型,支持多种推理场景比对,帮助开发者快速定位模型精度问题。Gitee开源https://gitee.com/ascend/msit/blob/master/msit/docs/llm/%E5%B7%A5%E5%85%B7-%E8%87%AA%E5%8A%A8%E6%AF%94%E5%AF%B9%E5%8A%9F%E8%83%BD%E4%BD%BF%E7%94%A8%E8%AF%B4%E6%98%8E.md
算子精度预检工具(msit llm opcheck)opcheck提供加速库内置算子的单算子精度预检能力,检测加速库算子精度是否达标。Gitee开源https://gitee.com/ascend/msit/blob/master/msit/docs/llm/%E5%B7%A5%E5%85%B7-%E7%B2%BE%E5%BA%A6%E9%A2%84%E6%A3%80%E4%BD%BF%E7%94%A8%E8%AF%B4%E6%98%8E.md
异常检测工具(msit llm errcheck)提供对模型推理过程中出现异常的检测能力,支持 Atlas 800I A2 推理卡场景下的算子溢出检测,包括饱和模式和非饱和模式。Gitee开源https://gitee.com/ascend/msit/blob/master/msit/docs/llm/%E5%B7%A5%E5%85%B7-%E5%BC%82%E5%B8%B8%E6%A3%80%E6%B5%8B%E4%BD%BF%E7%94%A8%E8%AF%B4%E6%98%8E.md
精度调试工具用于比对自有实现的算子运算结果与业界标准算子运算结果存在的差异,帮助开发人员快速定位算子精度问题的工具。社区https://www.hiascend.com/document/detail/zh/mindstudio/70RC2/T&ITools/training&inferencetools/atlasaccuracy_16_0001.html
性能分析
om模型性能分析(msit benchmark)benchmark推理功能可以通过使用针对指定的推理模型运行推理程序,测试推理模型的性能(包括吞吐率、时延)Gitee开源https://gitee.com/ascend/msit/tree/master/msit/docs/benchmark
性能分析工具性能分析工具是用于分析运行在昇腾AI处理器上的App工程各个运行阶段的关键性能瓶颈,并提出针对性能优化的建议,最终实现产品的极致性能。社区https://www.hiascend.com/document/detail/zh/mindstudio/70RC2/T&ITools/training&inferencetools/atlasprofiling_16_0001.html
扩展功能
tensor操作工具(msit tensor-view)tensor-view工具提供了查看tensor的接口,源数据是dump后生成的bin文件。支持对tensor数据进行链式切片、转置操作。Gitee开源https://gitee.com/ascend/msit/tree/master/msit/docs/tensor_view
算子及模型速查工具算子和模型速查工具提供了查询当前版本CANN支持的模型和算子功能。社区https://www.hiascend.com/document/detail/zh/mindstudio/70RC2/T&ITools/training&inferencetools/atlasopmodelquery_16_0001.html

本帖最后由 匿名用户2024/10/22 15:24:24 编辑

我要发帖子