From c1e63a5d2d7653f71f228bafa4246ccb48a79788 Mon Sep 17 00:00:00 2001 From: Andreas Kloeckner Date: Tue, 26 Jun 2018 23:15:23 -0500 Subject: [PATCH] Add an MPI example --- examples/wave/wave-min-mpi.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/examples/wave/wave-min-mpi.py b/examples/wave/wave-min-mpi.py index 26d22226..04d0b8a3 100644 --- a/examples/wave/wave-min-mpi.py +++ b/examples/wave/wave-min-mpi.py @@ -122,6 +122,8 @@ def main(write_output=True, order=4): from time import time t_last_step = time() + rank = comm.Get_rank() + for event in dt_stepper.run(t_end=final_t): if isinstance(event, dt_stepper.StateComputed): assert event.component_id == "w" @@ -131,7 +133,11 @@ def main(write_output=True, order=4): print(step, event.t, norm(queue, u=event.state_component[0]), time()-t_last_step) if step % 10 == 0: - vis.write_vtk_file("fld-%04d.vtu" % step, + vis.write_vtk_file( + "fld-%03d-%04d.vtu" % ( + rank, + step, + ), [ ("u", event.state_component[0]), ("v", event.state_component[1:]), -- GitLab