文档
注册

简介

介绍如何使用MindStudio工具实现不同框架的算子开发。包括支持MindSpore、PyTorch、TensorFlow、Caffe、ONNX的TBE算子开发和支持TensorFlow、PyTorch、ONNX、MindSpore的AI CPU算子开发,其中MindSpore框架的算子目前仅支持昇腾910 AI处理器。此过程中,主要介绍MindStudio工具操作部分,对代码实现部分仅做基本介绍,如需了解更多代码开发技巧,请参见对应算子开发指南。

MindStudio工具提供了算子开发样例,配置完成Ascend-cann-toolkit开发套件包后,可从MindStudio安装目录下的“/MindStudio/plugins/ascend-foundation/samples/operator_demo_projects/quick_start”获取算子开发样例。样例请参见自定义算子样例

Ascend-cann-toolkit安装目录/ascend-toolkit/latest/opp/op_impl/built-in/ai_core/tbe/impl目录下为昇腾AI处理器支持的内置AI Core算子的实现文件,开发者在进行算子代码实现时可进行参考。

搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词