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