Fix in 2D for NCIGodfreyFilter

dpgrote · Jun 11, 2024 · d383e51 · d383e51
1 parent 0df1442
commit d383e51
Showing 1 changed file with 4 additions and 6 deletions.
diff --git a/Source/Filter/NCIGodfreyFilter.cpp b/Source/Filter/NCIGodfreyFilter.cpp
@@ -122,18 +122,16 @@ void NCIGodfreyFilter::ComputeStencils()
     h_stencil_z[0] /= 2._rt;
 
     stencil_x.resize(h_stencil_x.size());
+    Gpu::copyAsync(Gpu::hostToDevice,h_stencil_x.begin(),h_stencil_x.end(),stencil_x.begin());
 #  if defined(WARPX_DIM_3D)
     stencil_y.resize(h_stencil_y.size());
     stencil_z.resize(h_stencil_z.size());
+    Gpu::copyAsync(Gpu::hostToDevice,h_stencil_y.begin(),h_stencil_y.end(),stencil_y.begin());
+    Gpu::copyAsync(Gpu::hostToDevice,h_stencil_z.begin(),h_stencil_z.end(),stencil_z.begin());
 #  elif (AMREX_SPACEDIM == 2)
     // In 2D, the filter applies stencil_y to the 2nd dimension
     stencil_y.resize(h_stencil_z.size());
-#  endif
-
-    Gpu::copyAsync(Gpu::hostToDevice,h_stencil_x.begin(),h_stencil_x.end(),stencil_x.begin());
-    Gpu::copyAsync(Gpu::hostToDevice,h_stencil_y.begin(),h_stencil_y.end(),stencil_y.begin());
-#  if defined(WARPX_DIM_3D)
-    Gpu::copyAsync(Gpu::hostToDevice,h_stencil_z.begin(),h_stencil_z.end(),stencil_z.begin());
+    Gpu::copyAsync(Gpu::hostToDevice,h_stencil_z.begin(),h_stencil_z.end(),stencil_y.begin());
 #  endif
 
     Gpu::synchronize();