Scalable multi-GPU 3-D FFT for TSUBAME 2.0 supercomputer

  • Authors:
  • Akira Nukada;Kento Sato;Satoshi Matsuoka

  • Affiliations:
  • Tokyo Institute of Technology and Japan Science and Technology Agency;Tokyo Institute of Technology;Tokyo Institute of Technology and National Institute of Informatics and Japan Science and Technology Agency

  • Venue:
  • SC '12 Proceedings of the International Conference on High Performance Computing, Networking, Storage and Analysis
  • Year:
  • 2012

Quantified Score

Hi-index 0.00

Visualization

Abstract

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.