C++ extensions插件提供了将自定义算子映射到昇腾AI处理器的功能,为使用PyTorch框架的开发者提供了便捷的NPU算子库调用能力,基于PyTorch原生提供的自定义算子扩展功能,用户可以编译安装自定义算子wheel包并运行。本章节基于C++ extensions的方式介绍如何在昇腾NPU上完成自定义算子开发和适配。PyTorch官方的C++ extensions功能具体可参考。
此目录下展示了C++ extensions构建自定义算子库的所有样例。
[object Object]
提供了六种适配样例以供用户选择,用户可根据需要选择对应适配方法。
[object Object]