Overview

This section describes how to use MindStudio IDE to develop custom operators under different frameworks.

For example, develop TBE operators under MindSpore, PyTorch, TensorFlow, Caffe, and ONNX frameworks and develop AI CPU operators under MindSpore, TensorFlow, PyTorch, and ONNX frameworks. Currently, MindSpore operators apply only to Ascend 910 AI Processors.

This section mainly focuses on MindStudio IDE operations but also touches on important code implementations. For details about the programming, see the TBE & AI CPU Operator Development Guide.

After the Ascend-CANN-Toolkit configuration is complete, you can obtain an operator development sample from Ascend-CANN-Toolkit installation directory/ascend-toolkit/latest/tools/msopgen/template/operator_demo_projects. For details, see Importing an Operator Project.

The Ascend-CANN-Toolkit installation directory/ascend-toolkit/latest/opp/built-in/op_impl/ai_core/tbe/impl directory stores the implementation files of the built-in AI Core operators supported by the Ascend AI Processor. You can refer to these files when implementing the operator code.