https://github.com/zzudongxiang/astrasim

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

Last synced: about 1 month ago
JSON representation

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

• Host: GitHub
• URL: https://github.com/zzudongxiang/astrasim
• Owner: zzudongxiang
• Created: 2024-06-04T15:28:31.000Z (7 months ago)
• Default Branch: main
• Last Pushed: 2024-06-25T10:12:58.000Z (6 months ago)
• Last Synced: 2024-06-25T17:18:00.798Z (6 months ago)
• Homepage: https://gitee.com/zzudongxiang/astrasim
• Size: 930 KB
• Stars: 2
• Watchers: 1
• Forks: 0
• Open Issues: 0
• Metadata Files:
  • Readme: README.md

Awesome Lists containing this project

README

        
# ASTRA-sim

**源码细节正在调试，暂未开放，欢迎邮件沟通 [[email protected]](mailto:[email protected])**

**源码细节正在调试，暂未开放，欢迎邮件沟通 [[email protected]](mailto:[email protected])**

**源码细节正在调试，暂未开放，欢迎邮件沟通 [[email protected]](mailto:[email protected])**

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

![astra_sim_overview](./images/astra_sim_overview.png)

## 1. 快速开始

- **项目的[安装](./docs/INSATLL.md)、[使用](./docs/MANUAL.md)、[帮助](./docs/HELP.md)等功能请参考[docs](./docs)文件夹中的说明文件**

  - **[INSTALL.md](./docs/INSTALL.md)** 项目的安装教程和安装指导，指示如何正确的构建所需的工具链

  - **[MANUAL.md](./docs/MANUAL.md)** 项目的使用帮助手册，指示如何使用本仓库进行ASTRA-sim仿真

  - **[HELP.md](./docs/HELP.md)** 关于项目开发过程中的一些帮助信息，包括参数解释、参数获取方法等

## 2. 仿真效果

### 2.1 Ascend910B 集合通信效果

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

![image-20240619192711266](./images/image-20240619192711266.png)

- **红色曲线** 为使用ASTRA-sim仿真的结果

- **蓝色曲线** 为使用Ascend 910B实测的实验数据