简介
介绍如何使用MindStudio工具实现不同框架的算子开发。
包括支持MindSpore、PyTorch、TensorFlow、Caffe、ONNX的TBE算子开发和支持TensorFlow、PyTorch、ONNX、MindSpore的AI CPU算子开发,其中MindSpore框架的算子目前仅支持Atlas 训练系列产品。
本章节主要介绍MindStudio工具操作部分,对代码实现部分仅做基本介绍,如需了解更多代码开发技巧,请参见对应《TBE&AI CPU算子开发指南》。
配置完成Ascend-cann-toolkit开发套件包后,可从Ascend-cann-toolkit安装目录/ascend-toolkit/latest/tools/msopgen/template/operator_demo_projects/获取算子开发样例。样例请参见导入算子工程样例。
Ascend-cann-toolkit安装目录/ascend-toolkit/latest/opp/built-in/op_impl/ai_core/tbe/impl目录下为昇腾AI处理器支持的内置AI Core算子的实现文件,开发者在进行算子代码实现时可进行参考。
父主题: 算子开发