开发者
下载

TorchNPU

作为Ascend for PyTorch社区的核心组件,TorchNPU是昇腾专为PyTorch打造的深度学习适配框架,使PyTorch框架能够直接调用昇腾NPU,为开发者提供强劲的AI算力支撑。

TorchNPU 26.2.X的关键特性

  • 支持alltoallvc通信算子,优化alltoall通信性能。
  • 支持通信域以分级建链方式初始化,性能优化。
  • 支持Inductor基于AscendC IR的自动算子融合。
  • 继承DeviceAllocator,支持在Eager模式下启用accelerator memory和graph相关接口。
  • 从本版本起,Inductor后端中的Torch-MLIR编译器已废弃,涉及《Torch.compile》和《环境变量》,具体变更内容请参考版本配套文档。
  • 从本版本起,基于毕昇编译器提供的编译优化功能已废弃,可以使用Clang进行替换。