From f1fcac8a018c46fe0ae587469b2bf92862110f7a Mon Sep 17 00:00:00 2001
From: Michael E Rowan <38045958+mrowan137@users.noreply.github.com>
Date: Tue, 5 May 2020 17:40:24 -0700
Subject: Switch costs to LayoutData (#970)

Cleanup

Cleanup
---
 Source/Evolve/WarpXEvolve.cpp | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

(limited to 'Source/Evolve/WarpXEvolve.cpp')

diff --git a/Source/Evolve/WarpXEvolve.cpp b/Source/Evolve/WarpXEvolve.cpp
index bc1e31806..8ca059184 100644
--- a/Source/Evolve/WarpXEvolve.cpp
+++ b/Source/Evolve/WarpXEvolve.cpp
@@ -61,7 +61,7 @@ WarpX::Evolve (int numsteps)
         if (warpx_py_beforestep) warpx_py_beforestep();
 #endif
 
-        amrex::Vector<amrex::Real>* cost = WarpX::getCosts(0);
+        amrex::LayoutData<amrex::Real>* cost = WarpX::getCosts(0);
         if (cost) {
 #ifdef WARPX_USE_PSATD
             amrex::Abort("LoadBalance for PSATD: TODO");
@@ -82,7 +82,7 @@ WarpX::Evolve (int numsteps)
                     // (Giving more importance to most recent costs; only needed
                     // for timers update, heuristic load balance considers the
                     // instantaneous costs)
-                    for (int i=0; i<cost->size(); ++i)
+                    for (int i : cost->IndexArray())
                     {
                         (*cost)[i] *= (1. - 2./load_balance_intervals.localPeriod(step+1));
                     }
-- 
cgit v1.2.3