跳转至

简介

MindStudio Ops Generator(算子工程工具,msOpGen)是算子开发效率提升工具,集成于 MindStudio 工具链中。完成算子分析和原型定义后,可使用 msOpGen 工具生成自定义算子工程,并进行编译部署。

核心功能

  • 算子工程生成msopgen gen):基于算子原型定义 JSON 文件,自动生成完整的算子工程框架(含 Host 侧和 Kernel 侧代码模板、编译脚本等)
  • 仿真流水图解析msopgen sim):解析性能仿真环境生成的 dump 数据,生成可在 Chrome tracing 中查看的算子仿真流水图文件

配套工具

  • msOpST(算子测试工具):在真实硬件环境中对算子进行 ST(System Test)测试,验证算子功能正确性
  • msProf(性能分析工具) / msSanitizer(异常检测工具) / msDebug(调试工具):与 msOpGen 生成的算子工程深度集成,覆盖性能调优、内存检测和上板调试场景

适用场景

  • Ascend C 自定义算子开发
  • TBE / AI CPU 算子工程搭建
  • 多框架适配(TensorFlow、PyTorch、MindSpore、ONNX)