mpi4py/mpi4py-fft

mpi4py-fft with Apple Silicon M1/M2?

pancetta opened this issue · 4 comments

Looks like mpi4py-fft cannot easily be installed on an M1 or M2 arm-based architecture. Using conda does not work, because only recipes for osx-64 and linux-64 are given. Using pip fails (for me) with

      [...]
      Error compiling Cython file:
      ------------------------------------------------------------
      ...
      cimport fftwf_xfftn
              ^
      ------------------------------------------------------------
      
      mpi4py_fft/fftw/fftwf_xfftn.pyx:1:8: 'fftwf_xfftn.pxd' not found
      [...]

Any ideas?

Hi,
I guess we simply need to add osx-arm64 platform on the conda feedstock. You are more than welcome to try a PR to this end over there:-)

@pancetta I just tried to add this in a PR and the error message you refer to here is simply due to using cython version 3.0.0, which was released a month ago. I rolled back to cython < 1.0, but then I got a new arm-related error, where I'm stuck. Perhaps @dalcinl will know, because mpi4py seems to be building with osx-arm64....:-)

Hey @mikaem, thanks for trying!

Including osx-arm64 taken care of here.
Update to cython 3 going on here: #29