Skip to content

ASTRA-sim 是一个分布式机器学习系统模拟器。它可以系统地研究现代深度学习系统所面临的挑战,探索瓶颈问题,并为未来不同平台上开发大型 DNN 模型提供高效的方法。

Notifications You must be signed in to change notification settings

zzudongxiang/astrasim

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

43 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

ASTRA-sim

源码细节正在调试,暂未开放,欢迎邮件沟通 zzudongxiang@163.com

源码细节正在调试,暂未开放,欢迎邮件沟通 zzudongxiang@163.com

源码细节正在调试,暂未开放,欢迎邮件沟通 zzudongxiang@163.com

ASTRA-sim 是一个分布式机器学习系统模拟器。它可以系统地研究现代深度学习系统所面临的挑战,探索瓶颈问题,并为未来不同平台上开发大型 DNN 模型提供高效的方法。

astra_sim_overview

1. 快速开始

  • 项目的安装使用帮助等功能请参考docs文件夹中的说明文件
    • INSTALL.md 项目的安装教程和安装指导,指示如何正确的构建所需的工具链
    • MANUAL.md 项目的使用帮助手册,指示如何使用本仓库进行ASTRA-sim仿真
    • HELP.md 关于项目开发过程中的一些帮助信息,包括参数解释、参数获取方法等

2. 仿真效果

2.1 Ascend910B 集合通信效果

在华为Ascend 910B设备上进行8个NPU的集合通信仿真结果对比如下

image-20240619192711266

  • 红色曲线 为使用ASTRA-sim仿真的结果
  • 蓝色曲线 为使用Ascend 910B实测的实验数据

About

ASTRA-sim 是一个分布式机器学习系统模拟器。它可以系统地研究现代深度学习系统所面临的挑战,探索瓶颈问题,并为未来不同平台上开发大型 DNN 模型提供高效的方法。

Resources

Stars

Watchers

Forks

Packages

No packages published