diff --git a/fluidsim/operators/operators2d.py b/fluidsim/operators/operators2d.py index 001f3e3347b503d00eedd8e4b9b960459908f220_Zmx1aWRzaW0vb3BlcmF0b3JzL29wZXJhdG9yczJkLnB5..7896cf429972e866e576f32fb548b49c74680cce_Zmx1aWRzaW0vb3BlcmF0b3JzL29wZXJhdG9yczJkLnB5 100644 --- a/fluidsim/operators/operators2d.py +++ b/fluidsim/operators/operators2d.py @@ -73,8 +73,13 @@ return inc_var -nb_proc = mpi.nb_proc -rank = mpi.rank +if not fp.is_transpiling: + nb_proc = mpi.nb_proc + rank = mpi.rank +else: + nb_proc = 1 + rank = 0 + if nb_proc > 1: MPI = mpi.MPI comm = mpi.comm diff --git a/setup.py b/setup.py index 001f3e3347b503d00eedd8e4b9b960459908f220_c2V0dXAucHk=..7896cf429972e866e576f32fb548b49c74680cce_c2V0dXAucHk= 100644 --- a/setup.py +++ b/setup.py @@ -69,4 +69,5 @@ "fluiddyn.io", "fluiddyn.util", "fluiddyn.util.paramcontainer", + "fluiddyn.util.mpi", "fluiddyn.output", @@ -72,4 +73,6 @@ "fluiddyn.output", + "fluiddyn.calcul", + "fluiddyn.calcul.setofvariables", "fluidfft.fft2d.operators", "fluidfft.fft3d.operators", ),