Fix sumToLine for multiple components (#4266)

* Fix comment. * Fix the size of the device vector.
AMReX-Codes · Dec 16, 2024 · bdb4be3 · bdb4be3
1 parent 7111355
commit bdb4be3
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/Src/Base/AMReX_MultiFabUtil.H b/Src/Base/AMReX_MultiFabUtil.H
@@ -334,7 +334,7 @@ namespace amrex
      * Return a HostVector that contains the sum of the given MultiFab data in the plane
      * with the given normal direction.  The size of the vector is
      * domain.length(direction) x ncomp.  The vector is actually a 2D array, where the
-     * element for component icomp at spatial index k is at [icomp*ncomp+k].
+     * element for component icomp at spatial index k is at [icomp+ncomp*k].
      *
      * \param mf MultiFab data for summing
      * \param icomp starting component

diff --git a/Src/Base/AMReX_MultiFabUtil.cpp b/Src/Base/AMReX_MultiFabUtil.cpp
@@ -822,7 +822,7 @@ namespace amrex
 #ifdef AMREX_USE_GPU
         if (Gpu::inLaunchRegion())
         {
-            Gpu::DeviceVector<Real> dv(domain.length(direction), Real(0.0));
+            Gpu::DeviceVector<Real> dv(hv.size(), Real(0.0));
             Real* p = dv.data();
 
             for (MFIter mfi(mf); mfi.isValid(); ++mfi) {