Ascend Extension for PyTorch是个what？

1 Ascend Extension for PyTorch

Ascend Extension for PyTorch 插件是基于昇腾的深度学习适配框架，使昇腾NPU可以支持PyTorch框架，为PyTorch框架的使用者提供昇腾AI处理器的超强算力。

项目源码地址请参见Ascend/Pytorch。

昇腾为基于昇腾处理器和软件的行业应用及服务提供全栈AI计算基础设施。您可以通过访问昇腾社区，了解关于昇腾的更多信息。

2 为什么需要此扩展？

PyTorch本身是个模型训练和推理的AI框架，与华为的昇思（mindspore）和百度的飞浆（paddlepaddle）等是同样的产品形态。pytorch作为业界的顶流，无论是在业界，还是高校或者说是AI相关社区，其重要性都是无法撼动的。但是，pytorch本身只对cpu和gpu进行了较好的适配，这就导致了新出现的npu硬件产品，如果要在pytorch上进行无缝对接，肯定得进行一些适配。比如：如何调用新npu硬件上配套的高性能算子？

3 Ascend在pytorch上的对接方案

以下架构图来源于昇腾社区。
As