Scalable multi-GPU 3-D FFT for TSUBAME 2.0 Supercomputer
Scalable multi-GPU 3-D FFT for TSUBAME 2.0 Supercomputer
复制标题
DOI:
10.1109/sc.2012.100
复制
发表时间:
2012-11
期刊:
影响因子:
--
通讯作者:
Akira Nukada;Kento Sato;S. Matsuoka
中科院分区:
文献类型:
--
作者:
Akira Nukada;Kento Sato;S. Matsuoka
For scalable 3-D FFT computation using multiple GPUs, efficient all-to-all communication between GPUs is the most important factor in good performance. Implementations with point-to-point MPI library functions and CUDA memory copy APIs typically exhibit very large overheads especially for small message sizes in all-to-all communications between many nodes. We propose several schemes to minimize the overheads, including employment of lower-level API of InfiniBand to effectively overlap intra- and inter-node communication, as well as auto-tuning strategies to control scheduling and determine rail assignments. As a result we achieve very good strong scalability as well as good performance, up to 4.8TFLOPS using 256 nodes of TSUBAME 2.0 Supercomputer (768 GPUs) in double precision.