wave-op-mpi: add simple timing (#197)

* wave-op-mpi: add simple timing

* fix

Author: matthiasdiener

examples/wave/wave-op-mpi.py

@@ -224,6 +224,9 @@ def rhs(t, w):
     if comm.rank == 0:
         logger.info("dt = %g", dt)
 
+    import time
+    start = time.time()
+
     t = 0
     t_final = 3
     istep = 0
@@ -236,6 +239,7 @@ def rhs(t, w):
         l2norm = actx.to_numpy(op.norm(dcoll, fields.u, 2))
 
         if istep % 10 == 0:
+            stop = time.time()
             linfnorm = actx.to_numpy(op.norm(dcoll, fields.u, np.inf))
             nodalmax = actx.to_numpy(op.nodal_max(dcoll, "vol", fields.u))
             nodalmin = actx.to_numpy(op.nodal_min(dcoll, "vol", fields.u))
@@ -244,7 +248,8 @@ def rhs(t, w):
                             f"L2: {l2norm} "
                             f"Linf: {linfnorm} "
                             f"sol max: {nodalmax} "
-                            f"sol min: {nodalmin}")
+                            f"sol min: {nodalmin} "
+                            f"wall: {stop-start} ")
             if visualize:
                 vis.write_parallel_vtk_file(
                     comm,
@@ -254,6 +259,7 @@ def rhs(t, w):
                         ("v", fields.v),
                     ]
                 )
+            start = stop
 
         t += dt
         istep += 1