account for ghost elements

Co-authored-by: Jeremy E Kozdon <jekozdon@nps.edu>
CliMA · Jun 30, 2020 · 1fbfd6e · 1fbfd6e
1 parent 02f8062
commit 1fbfd6e
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/Ocean/HydrostaticBoussinesq/HydrostaticBoussinesqModel.jl b/src/Ocean/HydrostaticBoussinesq/HydrostaticBoussinesqModel.jl
@@ -640,10 +640,10 @@ function update_auxiliary_state_gradient!(
     wz0 = A.data[:, index_wz0, :]
 
     # project w(z=0) down the stack
-    Nq, Nqk, _, _, nelemv, _, nelemh, _ = basic_grid_info(dg)
+    Nq, Nqk, _, _, nelemv, nelemh, nhorzrealelem, _ = basic_grid_info(dg)
     data = reshape(A.data, Nq^2, Nqk, number_auxiliary, nelemv, nelemh)
-    flat_wz0 = @view data[:, end:end, index_w, end:end, :]
-    boxy_wz0 = @view data[:, :, index_wz0, :, :]
+    flat_wz0 = @view data[:, end:end, index_w, end:end, 1:nhorzrealelem]
+    boxy_wz0 = @view data[:, :, index_wz0, :, 1:nhorzrealelem]
     boxy_wz0 .= flat_wz0
 
     return true