昇腾社区首页
中文
注册

新增特性

  • Ascend Extension for PyTorch(即torch-npu)2.1.0及以上版本支持sparse COO格式稀疏tensor创建,支持相关的基础辅助接口,以及部分算子计算。
  • Ascend Extension for PyTorch主动报错场景增加错误码打印,辅助大模型场景下快速定位出错设备和报错模块。
  • 新增支持图模式训练部分特性,具体可参考PyTorch 图模式使用指南(TorchAir)
  • 新增部分API支持及特性支持,具体可参考API 参考
  • Ascend Extension for PyTorch 2.1.0及以上版本支持单进程多卡。