Comprehensive Deadlock Prevention for GPU Collective Communication
Distributed deep neural network training necessitates efficient GPU collective communications, which are inherently susceptible to deadlocks. GPU collective deadlocks arise easily in distributed deep learning applications when multiple collectives circularly wait for each other. GPU collective deadlocks pose a significant challenge to the correct functioning and efficiency of distributed deep learning, and no general effective solutions are currently available. Only in specific scenarios, ad-hoc methods, making an application invoke collectives in a consistent order across GPUs, can be used to prevent circular collective dependency and deadlocks.
Authors
- Pengze Li (ORCID: https://orcid.org/0000-0001-7015-0491)
- Yongquan Fu (ORCID: https://orcid.org/0000-0002-7564-5239)
- Rongkai Zhang (ORCID: https://orcid.org/0009-0003-1652-7202)
- Lichen Pan (ORCID: https://orcid.org/0000-0001-7451-0140)
- Yuan Jinhui (ORCID: https://orcid.org/0000-0002-0700-2645)
- Zhen Xiao (ORCID: https://orcid.org/0000-0002-6784-9709)
- Juncheng Liu (ORCID: https://orcid.org/0000-0002-8213-7718)
Institutions
- National University of Defense Technology (CN)
- Peking University (CN)
- InflowControl (Norway) (NO)
- Qi2 (US)
Publication Details
- Published
- 2025-03-26
- DOI
- https://doi.org/10.1145/3689031.3717466
- Citations
- 1
- Primary Topic
- Advanced Memory and Neural Computing
- Type
- article
- Field-Weighted Citation Impact
- 0.73