Efficient Process Arrival Pattern Aware Collective Communication for Deep Learning
Efficient Process Arrival Pattern Aware Collective Communication for Deep Learning
复制标题
用于深度学习的高效流程到达模式感知集体通信
DOI:
--
复制
发表时间:
2022
期刊:
影响因子:
--
通讯作者:
A. Afsahi
中科院分区:
文献类型:
--
作者:
Pedram Alizadeh;A. Sojoodi;Y. H. Temuçin;A. Afsahi
MPI collective communication operations are used extensively in parallel applications. As such, researchers have been investigating how to improve their performance and scalability to directly impact application performance. Unfortunately, most of these studies are based on the premise that all processes arrive at the collective call simultaneously. A few studies though have shown that imbalanced Process Arrival Pattern (PAP) is ubiquitous in real environments, significantly affecting the collective performance. Therefore, devising PAP-aware collective algorithms that could improve performance, while challenging, is highly desirable. This paper is along those lines but in the context of Deep Learning (DL) workloads that have become maintstream. This paper presents a brief characterization of collective communications, in particular MPI_Allreduce, in the Horovod distributed Deep Learning framework and shows that the arrival pattern of MPI processes is indeed imbalanced. It then proposes an intra-node shared-memory PAP-aware MPI_Allreduce algorithm for small to medium messages, where the leader process is dynamically chosen based on the arrival time of the processes at each invocation of the collective call. We then propose an intra-node PAP-aware algorithm for large messages that dynamically constructs the reduction schedule at each MPI_Allreduce invocation. Finally, we propose a PAP-aware cluster-wide hierarchical algorithm, which is extended by utilizing our intra-node PAP-aware designs, that imposes less data dependency among processes given its hierarchical nature compared to flat algorithms. The proposed algorithms deliver up to 58% and 17% improvement at the micro-benchmark and Horovod with TensorFlow application over the native algorithms, respectively.
DOI:
10.1145/3392717.3392771
发表时间:
2020-06
期刊:
Proceedings of the 34th ACM International Conference on Supercomputing
影响因子:
--
作者:
通讯作者:
--
DOI:
10.1109/exampi52011.2020.00007
发表时间:
2020
期刊:
ExaMPI2020 - Workshop on Exascale MPI 2020
影响因子:
--
作者:
Ramesh, Bharath;Suresh, Kaushik Kandadi;Sarkauskas, Nick;Bayatpour, Mohammadreza;Hashmi, Jahanzeb Maqbool;Subramoni, Hari;Panda, Dhabaleswar K.
通讯作者:
Panda, Dhabaleswar K.