mpi4py/mpi4py-fft

Example reproduction error

ALWXAE opened this issue · 1 comments

Hello everyone,

I tried out your great package and tested this example (https://github.com/mpi4py/mpi4py-fft/blob/master/examples/darray.py) with minor attachment to print out s0,s1,s2:

darray.txt

I've tried to run the example code on a total of 1-8 nodes with python 3.11.6, but the code runs successfully only in the case of 1, 2, 3 and 5 nodes, otherwise error assertion arises:
assertion_error.txt

Attached are output of different processor numbers:
nproc1.txt
nproc2.txt
nproc3.txt
nproc4.txt
nproc5.txt
nproc6.txt
nproc7.txt
nproc8.txt

As you can see, the output changes for certain processor numbers, but I am unsure why. Help would be appreciated.
Regards
Hanru

Hi
Sorry about the late reply. I cannot seem to reproduce this. I have no errors for 6 nodes or less. For 7 I get an error, but that is because one of the arrays is of size (6, 6, 6).