%0 Conference Paper
%A Baumann, Thomas
%A Speck, Robert
%T Porting mpi4py-fft to GPU
%M FZJ-2024-04850
%D 2024
%X The mpi4py-fft library enables distributed fast Fourier transforms on CPUs with an easy to use interface and scales very well. We attempt to port this to GPUs, which significantly outperform the CPU counterpart at a given node count. While the porting is straightforward for the most part, the best communication strategy is still an open question for us.The algorithm relies on MPI alltoallw. Even with CUDA-aware MPI, this exhibits very poor performance on the Juelich computers. By replacing it with a custom communication strategy, throughput can be increased at a slight loss of generality. We would like to discuss optimising the strategy, or even if the performance of alltoallw can be increased by some measure.
%B 16th JLESC Workshop
%C 16 Apr 2024 - 18 Apr 2024, Kobe (Japan)
Y2 16 Apr 2024 - 18 Apr 2024
M2 Kobe, Japan
%F PUB:(DE-HGF)6
%9 Conference Presentation
%R 10.34734/FZJ-2024-04850
%U https://juser.fz-juelich.de/record/1028864