From 822d85bd6fe42c3ff2261578800c9d22eccf0cf2 Mon Sep 17 00:00:00 2001
From: Hannah Klion <hannah.klion@gmail.com>
Date: Wed, 13 Dec 2023 09:58:59 -0800
Subject: [PATCH] set local vars that can go on device

---
 Source/TimeIntegration/ROMSX_rho_eos.cpp | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git a/Source/TimeIntegration/ROMSX_rho_eos.cpp b/Source/TimeIntegration/ROMSX_rho_eos.cpp
index a564ad50..8385ab97 100644
--- a/Source/TimeIntegration/ROMSX_rho_eos.cpp
+++ b/Source/TimeIntegration/ROMSX_rho_eos.cpp
@@ -43,11 +43,16 @@ ROMSX::rho_eos (const Box& bx,
 //  equation of state.
 //-----------------------------------------------------------------------
 //
+    Real R0 = solverChoice.R0;
+    Real S0 = solverChoice.S0;
+    Real T0 = solverChoice.T0;
+    Real Tcoef = solverChoice.Tcoef;
+    Real Scoef = solverChoice.Scoef;
     amrex::ParallelFor(bx,
     [=] AMREX_GPU_DEVICE (int i, int j, int k)
     {
-        rho(i,j,k)  = solverChoice.R0 - solverChoice.R0*solverChoice.Tcoef*(temp(i,j,k,nrhs)-solverChoice.T0);
-        rho(i,j,k) += solverChoice.R0*solverChoice.Scoef*(salt(i,j,k,nrhs)-solverChoice.S0) - 1000.0_rt;
+        rho(i,j,k)  = R0 - R0*Tcoef*(temp(i,j,k,nrhs)-T0);
+        rho(i,j,k) += R0*Scoef*(salt(i,j,k,nrhs)-S0) - 1000.0_rt;
     });
 
 //