20 #pragma omp parallel if (Gpu::notInLaunchRegion())
22 for ( MFIter mfi(*(vel_t_avg),TilingIfNotGPU()); mfi.isValid(); ++mfi)
25 Box tbx = mfi.tilebox();
28 const Array4<Real>& velx =
xvel.array(mfi);
29 const Array4<Real>& vely =
yvel.array(mfi);
30 const Array4<Real>& velz =
zvel.array(mfi);
33 Array4<Real> vel_t_avg_arr = vel_t_avg->array(mfi);
35 ParallelFor(tbx, [=] AMREX_GPU_DEVICE (
int i,
int j,
int k)
37 Real u_cc =
myhalf * ( velx(i,j,k) + velx(i+1,j ,k ) );
38 Real v_cc =
myhalf * ( vely(i,j,k) + vely(i ,j+1,k ) );
39 Real w_cc =
myhalf * ( velz(i,j,k) + velz(i ,j ,k+1) );
40 Real umag_cc = std::sqrt(u_cc*u_cc + v_cc*v_cc + w_cc*w_cc);
41 vel_t_avg_arr(i,j,k,0) += u_cc * dt;
42 vel_t_avg_arr(i,j,k,1) += v_cc * dt;
43 vel_t_avg_arr(i,j,k,2) += w_cc * dt;
44 vel_t_avg_arr(i,j,k,3) += umag_cc * dt;
constexpr amrex::Real myhalf
Definition: ERF_Constants.H:11
ParallelFor(grown_box, [=] AMREX_GPU_DEVICE(int i, int j, int k) { qrcuten_arr(i, j, k)=Real(0);qscuten_arr(i, j, k)=Real(0);qicuten_arr(i, j, k)=Real(0);})
amrex::Real Real
Definition: ERF_ShocInterface.H:19
@ xvel
Definition: ERF_IndexDefines.H:175
@ zvel
Definition: ERF_IndexDefines.H:177
@ yvel
Definition: ERF_IndexDefines.H:176