mpiFFT4py performs FFTs in parallel in Python. It is developed to be able to do FFTs in parallel on a three-dimensional computational box (a structured grid), but there are also routines for doing the FFTs on a 2D mesh. The FFTs are computed using serial routines from either numpy.fft or pyfftw, whereas required MPI communications are performed using mpi4py.
bkmgit / mpifft4py Goto Github PK
View Code? Open in Web Editor NEWThis project forked from spectraldns/mpifft4py
Parallel FFT in 3D or 2D using MPI for Python. Slab or pencil decomposition possible in 3D.
License: GNU Lesser General Public License v3.0