框架适配开发
场景描述
框架适配开发,指CANN算子库中已实现了相应的算子,但未适配第三方框架的场景。可以查看《算子加速库接口参考》,判断算子库是否包含对应算子。
例如,算子库中已存在某算子,此算子已经适配了TensorFlow框架,但未适配ONNX框架,此种场景下,可直接进行ONNX框架的适配。
开发流程
框架适配场景下的开发流程如下图所示:
图1 算子框架适配开发流程
步骤 |
描述 |
参考 |
---|---|---|
环境准备 |
准备算子开发及运行验证所依赖的开发环境与运行环境。 |
|
工程创建 |
框架适配场景下,建议开发者直接基于已有算子工程追加进行算子适配插件的开发。 若无已有算子工程,您可以直接获取Ascend开源社区提供的算子样例,详细方法请参见右侧“参考”。
须知:
针对PyTorch框架的适配开发,无需进行工程创建。 |
|
算子适配 |
基于第三方框架(TensorFlow/Caffe/ONNX等)进行自定义算子开发的场景,开发人员需要进行插件的开发,将基于第三方框架的算子映射成适配昇腾AI处理器的算子。 |
|
算子网络测试 |
将自定义算子加载到网络模型中进行运行验证。 |
父主题: 选择算子开发场景