MindSpeed-Ops:基于昇腾生态的训练业务自定义算子项目

用户可用于昇腾平台训练业务的自定义算子实现与优化。项目提供Add、FusedCrossEntropyLoss等多种算子,基于triton技术,适配CANN 9.0.0、PyTorch 2.7.1等环境,助力提升训练性能。【此简介由AI生成】

分支7Tags0

项目介绍

用户可用于昇腾平台训练业务的自定义算子实现与优化。项目提供Add、FusedCrossEntropyLoss等多种算子,基于triton技术,适配CANN 9.0.0、PyTorch 2.7.1等环境,助力提升训练性能。【此简介由AI生成】

定制我的领域