昇腾社区首页
中文
注册

开发前准备

环境准备

安装CANN软件后,使用CANN运行用户编译、运行时,需要以CANN运行用户登录环境,执行. ${install_path}/set_env.sh命令设置环境变量。

  • “Ascend-cann-toolkit安装目录/ascend-toolkit/latest/opp/built-in/op_proto/inc”下提供了AICore和AICPU算子原型定义,用于通过算子原型构建Graph。
  • “Ascend-cann-toolkit安装目录/ascend-toolkit/latest/include/graph”下提供了Graph构建接口。
  • “Ascend-cann-toolkit安装目录/ascend-toolkit/latest/include/ge”下提供了Graph运行接口。
  • “Ascend-cann-toolkit安装目录/ascend-toolkit/latest/compiler/lib64/stub”下为相关依赖库。

网络分析

  • 如果通过算子原型构建Graph,需要根据原始网络,明确如下信息:
    1. 网络中包含哪些算子,以及这些算子的输入、输出、属性等信息。
    2. 网络中算子之间的关联关系。
    3. 确认原始网络中的算子在昇腾AI处理器是否支持,当前支持的算子请参考算子清单。如果不支持或不满足实际需要,请参考算子开发指南自定义算子,并将算子部署到硬件环境。
  • 如果是将原始模型解析为Graph,需要确认原始网络中的算子在昇腾AI处理器是否支持,当前支持的算子请参考算子清单。如果不支持或不满足实际需要,请参考算子开发指南自定义算子,并将算子部署到硬件环境。