HammingMesh: A Network Topology for Large-Scale Deep Learning
Numerous microarchitectural optimizations unlocked tremendous processing power for deep neural networks that in turn fueled the ongoing AI revolution. With the exhaustion of such optimizations, the growth of modern AI is now gated by the performance of training systems, especially their data movement. Instead of focusing on single accelerators, we investigate data-movement characteristics of large-scale training at full system scale. Based on our workload analysis, we design HammingMesh, a novel network topology that provides high bandwidth at low cost with high job scheduling flexibility. Specifically, HammingMesh can support full bandwidth and isolation to deep learning training jobs with two dimensions of parallelism. Furthermore, it also supports high global bandwidth for generic traffic. Thus, HammingMesh will power future large-scale deep learning systems with extreme bandwidth requirements.
Authors
- Salvatore Di Girolamo (ORCID: https://orcid.org/0000-0003-2197-8860)
- Torsten Hoefler (ORCID: https://orcid.org/0000-0002-1333-9797)
- Marco Heddes
- Daniele De Sensi (ORCID: https://orcid.org/0000-0002-7244-639X)
- Miguel Castro (ORCID: https://orcid.org/0000-0002-1277-3604)
- Shigang Li (ORCID: https://orcid.org/0000-0003-0022-7865)
- Steve Scott
- Deepak Goel
- Tommaso Bonoto
Institutions
- Microsoft (United States) (US)
- ETH Zurich (CH)
- Microsoft (United Kingdom) (GB)
- Microsoft (Switzerland) (CH)
Publication Details
- Journal
- Communications of the ACM
- Published
- 2024-11-21
- DOI
- https://doi.org/10.1145/3623490
- Citations
- 1
- Primary Topic
- Parallel Computing and Optimization Techniques
- Type
- article
- Field-Weighted Citation Impact
- 0.49