自定义算子样例

MindStudio工具提供了自定义算子样例，涵盖Mindspore、TensorFlow、PyTorch等多种AI框架，供用户学习与参考。打开样例工程步骤如下。

1. 进入工程导入页面。
   • MindStudio欢迎界面：在“Welcome to MindStudio”窗口中，单击“Open”。
   • MindStudio工程界面：在顶部菜单栏单击“File > Open...” 。

2. 在“Open File or Project”窗口中，选择样例工程所在文件夹。

   样例工程所在路径为MindStudio安装路径下的“MindStudio/plugins/ascend-foundation/samples/operator_demo_projects/cann/”。

   cann目录结构如下，选择需要打开的样例工程，在样例工程下执行算子编译、部署等操作。

   ├── AICPU
   │   ├── Onnx
   │   └── Tensorflow
   ├── DSL
   │   ├── Mindspore
   │   ├── Onnx
   │   ├── PyTorch
   │   └── Tensorflow
   └── TIK
       ├── Mindspore
       ├── PyTorch
       └── Tensorflow

3. 单击“OK”，导入样例工程。

   弹出“Import Support Configuration”对话框，选择相应参数后，单击Create按钮。

   表1 “Import Support Configuration”对话框

   参数	值
   Type	项目类型。 Ascend Operator Ascend App Ascend Training 下拉框选择Ascend Operator。
   Framework	算子所在模型文件的框架类型。 MindSpore PyTorch TensorFlow Onnx 根据打开的样例工程框架选择。

• 更换CANN包时，算子样例工程中cann目录会被覆盖。如果用户有修改，请在更换CANN包之前进行备份。
• 自定义算子模板代码仅作为示例，供算子开发过程参考，不支持用户基于自定义算子模板进行产品化开发。
• Onnx算子编译时需要protoc版本为3.13.0。
• 目前在Windows版本下：
  • 自定义算子样例代码中可能会出现无法识别的语义被红色波浪线标出，但不影响正常开发流程，用户可忽略该提示。
  • 若UT测试运行配置信息中的Case Names列表为空。可能由于算子样例的UT测试代码中导入了te或tbe模块，该样例暂不支持UT测试。
• 在ST测试阶段，请参考3配置如下环境变量：

  ASCEND_HOME=/usr/local/Ascend/ascend-toolkit/latest
  ASCEND_OPP_PATH=${ASCEND_HOME}/opp