Explicitly gather as Array for bcast in visu

Author: luraess

scripts/stokes_3D/Stokes3D_ve_multixpu.jl

@@ -264,11 +264,11 @@ end
     if (me==0) @printf("Total iters = %d (%d steps), time = %1.3e sec (@ T_eff = %1.2f GB/s) \n", ittot, nt, wtime, round(T_eff, sigdigits=2)) end
     # Visualisation
     if do_viz || do_save_viz
-        Pt_inn .= inn(Pt);   gather!(Pt_inn, Pt_v)
-        Vz_inn .= av_zi(Vz); gather!(Vz_inn, Vz_v)
-        Rz_inn .= av_za(Rz); gather!(Rz_inn, Rz_v)
-        Mus_inn.= inn(Mus);  gather!(Mus_inn, Mus_v)
-        τxz_inn.= av_xza(τxz); gather!(τxz_inn, τxz_v)
+        Pt_inn .= Array(inn(Pt));   gather!(Pt_inn, Pt_v)
+        Vz_inn .= Array(av_zi(Vz)); gather!(Vz_inn, Vz_v)
+        Rz_inn .= Array(av_za(Rz)); gather!(Rz_inn, Rz_v)
+        Mus_inn.= Array(inn(Mus));  gather!(Mus_inn, Mus_v)
+        τxz_inn.= Array(av_xza(τxz)); gather!(τxz_inn, τxz_v)
         if me==0 && do_viz
             p1 = heatmap(Xi_g, Zi_g, Pt_v[:,y_sl,:]', aspect_ratio=1, xlims=(Xi_g[1],Xi_g[end]), zlims=(Zi_g[1],Zi_g[end]), c=:viridis, title="Pressure")
             p2 = heatmap(Xi_g, Zi_g, Vz_v[:,y_sl,:]', aspect_ratio=1, xlims=(Xi_g[1],Xi_g[end]), zlims=(Zi_g[1],Zi_g[end]), c=:viridis, title="Vz")

scripts/stokes_3D/Stokes3D_ve_multixpu_perf.jl

@@ -264,11 +264,11 @@ end
     if (me==0) @printf("Total iters = %d (%d steps), time = %1.3e sec (@ T_eff = %1.2f GB/s) \n", ittot, nt, wtime, round(T_eff, sigdigits=3)) end
     # Visualisation
     if do_viz || do_save_viz
-        Pt_inn .= inn(Pt);   gather!(Pt_inn, Pt_v)
-        Vz_inn .= av_zi(Vz); gather!(Vz_inn, Vz_v)
-        Rz_inn .= av_za(Rz); gather!(Rz_inn, Rz_v)
-        Mus_inn.= inn(Mus);  gather!(Mus_inn, Mus_v)
-        τxz_inn.= av_xza(τxz); gather!(τxz_inn, τxz_v)
+        Pt_inn .= Array(inn(Pt));   gather!(Pt_inn, Pt_v)
+        Vz_inn .= Array(av_zi(Vz)); gather!(Vz_inn, Vz_v)
+        Rz_inn .= Array(av_za(Rz)); gather!(Rz_inn, Rz_v)
+        Mus_inn.= Array(inn(Mus));  gather!(Mus_inn, Mus_v)
+        τxz_inn.= Array(av_xza(τxz)); gather!(τxz_inn, τxz_v)
         if me==0 && do_viz
             p1 = heatmap(Xi_g, Zi_g, Pt_v[:,y_sl,:]', aspect_ratio=1, xlims=(Xi_g[1],Xi_g[end]), zlims=(Zi_g[1],Zi_g[end]), c=:viridis, title="Pressure")
             p2 = heatmap(Xi_g, Zi_g, Vz_v[:,y_sl,:]', aspect_ratio=1, xlims=(Xi_g[1],Xi_g[end]), zlims=(Zi_g[1],Zi_g[end]), c=:viridis, title="Vz")

scripts/stokes_3D/previous_damp/Stokes3D0_multixpu.jl

@@ -225,9 +225,9 @@ end
     if (me==0) @printf("Total steps = %d, err = %1.3e, time = %1.3e sec (@ T_eff = %1.2f GB/s) \n", iter, err, wtime, round(T_eff, sigdigits=2)) end
     # Visualisation
     if do_viz
-        Pt_inn .= inn(Pt);   gather!(Pt_inn, Pt_v)
-        Vz_inn .= av_zi(Vz); gather!(Vz_inn, Vz_v)
-        Rz_inn .= av_za(Rz); gather!(Rz_inn, Rz_v)
+        Pt_inn .= Array(inn(Pt));   gather!(Pt_inn, Pt_v)
+        Vz_inn .= Array(av_zi(Vz)); gather!(Vz_inn, Vz_v)
+        Rz_inn .= Array(av_za(Rz)); gather!(Rz_inn, Rz_v)
         if (me==0)
             p1 = heatmap(Xi_g, Zi_g, Pt_v[:,y_sl,:]', aspect_ratio=1, xlims=(Xi_g[1],Xi_g[end]), zlims=(Zi_g[1],Zi_g[end]), c=:viridis, title="Pressure")
             p2 = heatmap(Xi_g, Zi_g, Vz_v[:,y_sl,:]', aspect_ratio=1, xlims=(Xi_g[1],Xi_g[end]), zlims=(Zi_g[1],Zi_g[end]), c=:viridis, title="Vz")