适配开发(PyTorch框架)

简介

PyTorch算子映射为TBE算子通过修改PyTorch框架进行实现。PyTorch框架适配修改的主要在ATen,C10,torch等文件夹,其中涉及的修改类型主要有C++类型,配置类型(yaml),ACL类型和依赖库,Python,CMake类型等。

PyTorch环境准备

  1. 安装PyTorch依赖。

    请参见PyTorch算子开发指南中的“算子开发准备"章节。

  2. 编译和安装PyTorch框架。

    请参见PyTorch算子开发指南中的“算子适配开发>编译和安装PyTorch框架"章节。

算子适配开发

PyTorch框架适配开发操作请参见PyTorch算子开发指南中的“算子适配开发"章节。