PyTorch算子映射为TBE算子通过修改PyTorch框架进行实现。PyTorch框架适配修改的主要在ATen,C10,torch等文件夹,其中涉及的修改类型主要有C++类型,配置类型(yaml),ACL类型和依赖库,Python,CMake类型等。
请参见《PyTorch算子开发指南》中的“算子开发准备"章节。
请参见《PyTorch算子开发指南》中的“算子适配开发>编译和安装PyTorch框架"章节。
PyTorch框架适配开发操作请参见《PyTorch算子开发指南》中的“算子适配开发"章节。