22 #pragma omp parallel if (Gpu::notInLaunchRegion())
24 for ( MFIter mfi(*(vel_t_avg),TilingIfNotGPU()); mfi.isValid(); ++mfi)
27 Box tbx = mfi.tilebox();
30 const Array4<Real>& velx =
xvel.array(mfi);
31 const Array4<Real>& vely =
yvel.array(mfi);
32 const Array4<Real>& velz =
zvel.array(mfi);
35 Array4<Real> vel_t_avg_arr = vel_t_avg->array(mfi);
37 ParallelFor(tbx, [=] AMREX_GPU_DEVICE (
int i,
int j,
int k)
39 Real u_cc =
myhalf * ( velx(i,j,k) + velx(i+1,j ,k ) );
40 Real v_cc =
myhalf * ( vely(i,j,k) + vely(i ,j+1,k ) );
41 Real w_cc =
myhalf * ( velz(i,j,k) + velz(i ,j ,k+1) );
42 Real umag_cc = std::sqrt(u_cc*u_cc + v_cc*v_cc + w_cc*w_cc);
43 vel_t_avg_arr(i,j,k,0) += u_cc * dt;
44 vel_t_avg_arr(i,j,k,1) += v_cc * dt;
45 vel_t_avg_arr(i,j,k,2) += w_cc * dt;
46 vel_t_avg_arr(i,j,k,3) += umag_cc * dt;
constexpr amrex::Real myhalf
Definition: ERF_Constants.H:13
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:176
@ zvel
Definition: ERF_IndexDefines.H:178
@ yvel
Definition: ERF_IndexDefines.H:177