Fix norm logging in advection, Maxwell examples for device scalars

inducer · Jul 3, 2021 · 1c44c4b · 1c44c4b
1 parent 0d66c79
commit 1c44c4b
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 8 deletions.
diff --git a/examples/advection/surface.py b/examples/advection/surface.py
@@ -216,7 +216,7 @@ def rhs(t, u):
     dt_stepper = set_up_rk4("u", dt, u0, rhs)
     plot = Plotter(actx, dcoll, order, visualize=visualize)
 
-    norm_u = op.norm(dcoll, u0, 2)
+    norm_u = actx.to_numpy(op.norm(dcoll, u0, 2))
 
     step = 0
 
@@ -251,7 +251,7 @@ def rhs(t, u):
 
         step += 1
         if step % 10 == 0:
-            norm_u = op.norm(dcoll, event.state_component, 2)
+            norm_u = actx.to_numpy(op.norm(dcoll, event.state_component, 2))
             plot(event, "fld-surface-%04d" % step)
 
         logger.info("[%04d] t = %.5f |u| = %.5e", step, event.t, norm_u)

diff --git a/examples/advection/var-velocity.py b/examples/advection/var-velocity.py
@@ -216,7 +216,7 @@ def rhs(t, u):
             continue
 
         if step % 10 == 0:
-            norm_u = op.norm(dcoll, event.state_component, 2)
+            norm_u = actx.to_numpy(op.norm(dcoll, event.state_component, 2))
             plot(event, "fld-var-velocity-%04d" % step)
 
         step += 1

diff --git a/examples/advection/weak.py b/examples/advection/weak.py
@@ -186,7 +186,7 @@ def rhs(t, u):
             continue
 
         if step % 10 == 0:
-            norm_u = op.norm(dcoll, event.state_component, 2)
+            norm_u = actx.to_numpy(op.norm(dcoll, event.state_component, 2))
             plot(event, "fld-weak-%04d" % step)
 
         step += 1

diff --git a/examples/maxwell/cavities.py b/examples/maxwell/cavities.py
@@ -129,10 +129,10 @@ def norm(u):
             step += 1
             e, h = maxwell_operator.split_eh(event.state_component)
 
-            norm_e0 = norm(u=e[0])
-            norm_e1 = norm(u=e[1])
-            norm_h0 = norm(u=h[0])
-            norm_h1 = norm(u=h[1])
+            norm_e0 = actx.to_numpy(norm(u=e[0]))
+            norm_e1 = actx.to_numpy(norm(u=e[1]))
+            norm_h0 = actx.to_numpy(norm(u=h[0]))
+            norm_h1 = actx.to_numpy(norm(u=h[1]))
 
             logger.info(
                 "[%04d] t = %.5f |e0| = %.5e, |e1| = %.5e, |h0| = %.5e, |h1| = %.5e",