AI开发平台MODELARTS-使用Notebook开发Ascend算子:概述
概述
训练、推理场景下,使用第三方框架时遇到不支持的算子,需要自己开发;网络调优时,发现一些算子组合性能较低,需重新开发高性能算子替换低性能的算子,此时可以通过VS Code一键连接云上Notebook,使用云上资源,在VS Code端进行算子开发、调试等。Notebook中已经配置好环境,您无需进行CANN软件安装和环境变量配置就可以进行工程化开发。
本文档提供了一套算子工程样例代码,您可以直接使用。如果需要了解Ascend算子的编程模型等,请参见昇腾文档。Notebook中已经配置好环境,无需进行CANN软件安装和环境变量配置,直接在VS Code远端环境中直接进行算子分析及后续操作。