Skip to content

About

Learning resources and technical references for Ascend

Resources

Stars

7 stars

Watchers

0 watching

Forks

Latest commit

 

History

2 Commits

Folders and files

NameName
Last commit message
Last commit date
 
 
 
 

Repository files navigation

Ascend Dev References

English

昇腾(Ascend)训推开发中常用的学习与参考资料,覆盖推理、训练、kernel、算子 DSL、模板、性能分析和硬件系统,方便快速定位相关开源项目、工具与文档。

以下收录的 repo、文档和论文均为公开资料。

常用文档与最佳实践

推理

  • vLLM Ascend:vLLM 的 Ascend 硬件插件。
  • SGLang:已合入 Ascend NPU 支持的推理/服务框架。
  • Omni-Infer:面向 Ascend 的分布式推理加速与部署工具集。
  • CANN Recipes Infer:CANN 社区推理实践和可复用方案。
  • torchair:面向 Ascend 的 PyTorch torch.compile 与 GE 图模式编译支持。
  • xLLM:来自京东社区、面向 Ascend 的高性能大模型推理与服务框架。

训练

  • MindSpeed-LLM:Ascend 生态大语言模型训练套件,覆盖预训练、微调、权重转换、评估和推理。
  • MindSpeed-MM:Ascend 生态多模态大模型训练套件。
  • MindSpeed-RL:Ascend 生态强化学习训练套件。
  • verl Ascend Tutorial:verl 已合入的 Ascend 训练与强化学习支持教程。
  • VIME:支持 Ascend 的 RL 后训练框架。
  • slime-ascend:面向 Ascend 的 RL 训练项目。
  • TRL:Hugging Face 的大模型强化学习训练框架,已支持 Ascend。
  • DeepSpeed:面向大规模模型训练的系统优化框架,已支持 Ascend。

Kernel 算子库

算子 DSL 和模板

  • PyPTO:Parallel Tensor/Tile Operation 编程范式。
  • PTO-ISA:面向 tile 指令的虚拟 ISA 与工具链。
  • ATVOSS:基于 Ascend C 的 Vector 算子模板库。
  • CATLASS:面向高性能 GEMM/矩阵乘及相关融合算子的 CANN/Ascend 模板库。
  • TileLang-Ascend:面向 Ascend NPU 的 TileLang DSL 与编译后端。
  • Triton Ascend:面向 Ascend NPU 的 Triton DSL 与编译后端。

性能 Profiling

  • Ascend PyTorch Profiler:在 PyTorch 训练/推理脚本中通过 torch_npu.profiler 采集模型、CANN、NPU 算子及其 kernel 执行性能数据。
  • MindStudio Profiler / msProf:采集与解析 CANN 平台及昇腾 AI 处理器的软硬件性能数据。

Kernel 仿真

硬件与算力资源

  • CANNLab:在线申请和使用昇腾 NPU 算力,进行模型、算子和性能实验。

About

Learning resources and technical references for Ascend

Resources

Stars

7 stars

Watchers

0 watching

Forks

Releases

Packages

Contributors