Skip to content

Commit 0867211

Browse files
sbryngelsonsbryngelson
committed
add loop seq
1 parent e63dc1e commit 0867211

File tree

1 file changed

+22
-0
lines changed

1 file changed

+22
-0
lines changed

‎src/simulation/m_riemann_solvers.fpp‎

Lines changed: 22 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -4035,6 +4035,7 @@ contains
40354035

40364036
avg_vel(2) = 0.5_wp*(velL_vf(2)%sf(j_idx, k_idx, l_idx) + velR_vf(2)%sf(j_idx + 1, k_idx, l_idx)) ! v_r
40374037

4038+
!$acc loop seq
40384039
do i_loop = 1, 2 ! u_axial (1), u_radial (2)
40394040
dvel_avg_dy(i_loop) = 0.5_wp*(dvelL_dy_vf(i_loop)%sf(j_idx, k_idx, l_idx) + &
40404041
dvelR_dy_vf(i_loop)%sf(j_idx + 1, k_idx, l_idx)) ! du_i/dr
@@ -4049,6 +4050,7 @@ contains
40494050
! tau_zr shear stress: mu (dv_axial/dr + dv_r/dx_axial)
40504051
tau_Re(1, 2) = (dvel_avg_dy(1) + dvel_avg_dx(2))/Re_avg_rsx_vf(j_idx, k_idx, l_idx, 1)
40514052

4053+
!$acc loop seq
40524054
do i_loop = 1, 2 ! Axial (1) and radial (2) momentum components
40534055
flux_src_vf(contxe + i_loop)%sf(j_idx, k_idx, l_idx) = &
40544056
flux_src_vf(contxe + i_loop)%sf(j_idx, k_idx, l_idx) - tau_Re(1, i_loop)
@@ -4078,6 +4080,7 @@ contains
40784080
do k_idx = iy%beg, iy%end
40794081
do j_idx = ix%beg, ix%end
40804082

4083+
!$acc loop seq
40814084
do i_loop = 1, 3, 2 ! u_axial (1), u_azimuthal (3)
40824085
dvel_avg_dz(i_loop) = 0.5_wp*(dvelL_dz_vf(i_loop)%sf(j_idx, k_idx, l_idx) + &
40834086
dvelR_dz_vf(i_loop)%sf(j_idx + 1, k_idx, l_idx)) ! du_i/(r dtheta)
@@ -4091,6 +4094,7 @@ contains
40914094
! tau_z_theta shear stress: mu ( (1/r)dv_axial/dtheta + dv_theta/dx_axial )
40924095
tau_Re(1, 3) = (dvel_avg_dz(1)/y_cc(k_idx) + dvel_avg_dx(3))/Re_avg_rsx_vf(j_idx, k_idx, l_idx, 1)
40934096

4097+
!$acc loop seq
40944098
do i_loop = 1, 3, 2 ! Axial (1) and azimuthal (3) momentum components
40954099
flux_src_vf(contxe + i_loop)%sf(j_idx, k_idx, l_idx) = &
40964100
flux_src_vf(contxe + i_loop)%sf(j_idx, k_idx, l_idx) - tau_Re(1, i_loop)
@@ -4124,6 +4128,7 @@ contains
41244128
avg_vel(2) = 0.5_wp*(velL_vf(2)%sf(j_idx, k_idx, l_idx) + &
41254129
velR_vf(2)%sf(j_idx, k_idx + 1, l_idx)) ! v_r at interface k_idx+1/2
41264130
4131+
!$acc loop seq
41274132
do i_loop = 1, 2 ! u_axial (1), u_radial (2)
41284133
dvel_avg_dx(i_loop) = 0.5_wp*(dvelL_dx_vf(i_loop)%sf(j_idx, k_idx, l_idx) + &
41294134
dvelR_dx_vf(i_loop)%sf(j_idx, k_idx + 1, l_idx)) ! du_i/dx_axial
@@ -4172,6 +4177,7 @@ contains
41724177
avg_vel(3) = 0.5_wp*(velL_vf(3)%sf(j_idx, k_idx, l_idx) + &
41734178
velR_vf(3)%sf(j_idx, k_idx + 1, l_idx)) ! v_theta at interface
41744179
4180+
!$acc loop seq
41754181
do i_loop = 2, 3 ! u_radial (2), u_azimuthal (3)
41764182
dvel_avg_dz(i_loop) = 0.5_wp*(dvelL_dz_vf(i_loop)%sf(j_idx, k_idx, l_idx) + &
41774183
dvelR_dz_vf(i_loop)%sf(j_idx, k_idx + 1, l_idx)) ! du_i/(r dtheta)
@@ -4185,6 +4191,7 @@ contains
41854191
! tau_r_theta shear stress: mu ( (1/r)dv_r/dtheta - v_theta/r + dv_theta/dr )
41864192
tau_Re(2, 3) = ((dvel_avg_dz(2) - avg_vel(3))/y_cb(k_idx) + dvel_avg_dy(3))/Re_avg_rsy_vf(k_idx, j_idx, l_idx, 1)
41874193
4194+
!$acc loop seq
41884195
do i_loop = 2, 3 ! Radial (2) and azimuthal (3) momentum components
41894196
flux_src_vf(contxe + i_loop)%sf(j_idx, k_idx, l_idx) = &
41904197
flux_src_vf(contxe + i_loop)%sf(j_idx, k_idx, l_idx) - tau_Re(2, i_loop)
@@ -4214,19 +4221,25 @@ contains
42144221
do k_idx = iy%beg, iy%end
42154222
do j_idx = ix%beg, ix%end
42164223

4224+
!$acc loop seq
42174225
do i_loop = 2, 3 ! u_radial (2), u_azimuthal (3)
42184226
avg_vel(i_loop) = 0.5_wp*(velL_vf(i_loop)%sf(j_idx, k_idx, l_idx) + &
42194227
velR_vf(i_loop)%sf(j_idx, k_idx, l_idx + 1))
42204228
end do
42214229

4230+
!$acc loop seq
42224231
do i_loop = 1, 3, 2 ! u_axial (1), u_azimuthal (3)
42234232
dvel_avg_dx(i_loop) = 0.5_wp*(dvelL_dx_vf(i_loop)%sf(j_idx, k_idx, l_idx) + &
42244233
dvelR_dx_vf(i_loop)%sf(j_idx, k_idx, l_idx + 1)) ! du_i/dx_axial
42254234
end do
4235+
4236+
!$acc loop seq
42264237
do i_loop = 2, 3 ! u_radial (2), u_azimuthal (3)
42274238
dvel_avg_dy(i_loop) = 0.5_wp*(dvelL_dy_vf(i_loop)%sf(j_idx, k_idx, l_idx) + &
42284239
dvelR_dy_vf(i_loop)%sf(j_idx, k_idx, l_idx + 1)) ! du_i/dr_radial
42294240
end do
4241+
4242+
!$acc loop seq
42304243
do i_loop = 1, 3 ! All components
42314244
dvel_avg_dz(i_loop) = 0.5_wp*(dvelL_dz_vf(i_loop)%sf(j_idx, k_idx, l_idx) + &
42324245
dvelR_dz_vf(i_loop)%sf(j_idx, k_idx, l_idx + 1)) ! du_i/(r dtheta)_normal
@@ -4247,6 +4260,7 @@ contains
42474260
tau_Re(3, 2) = tau_Re(3, 2)/y_cc(k_idx)
42484261
tau_Re(3, 3) = tau_Re(3, 3)/y_cc(k_idx)
42494262

4263+
!$acc loop seq
42504264
do i_loop = 1, 3 ! Axial, radial, azimuthal momentum
42514265
flux_src_vf(contxe + i_loop)%sf(j_idx, k_idx, l_idx) = &
42524266
flux_src_vf(contxe + i_loop)%sf(j_idx, k_idx, l_idx) - tau_Re(3, i_loop)
@@ -4341,6 +4355,8 @@ contains
43414355
idx_right_phys(norm_dir) = idx_right_phys(norm_dir) + 1
43424356

43434357
vel_grad_avg = 0.0_wp
4358+
4359+
!$acc loop seq
43444360
do vel_comp_idx = 1, num_dims
43454361
vel_grad_avg(vel_comp_idx, 1) = 0.5_wp*(dvelL_dx_vf(vel_comp_idx)%sf(j_loop, k_loop, l_loop) + &
43464362
dvelR_dx_vf(vel_comp_idx)%sf(idx_right_phys(1), idx_right_phys(2), idx_right_phys(3)))
@@ -4355,6 +4371,7 @@ contains
43554371
end do
43564372

43574373
divergence_v = 0.0_wp
4374+
!$acc loop seq
43584375
do i_dim = 1, num_dims
43594376
divergence_v = divergence_v + vel_grad_avg(i_dim, i_dim)
43604377
end do
@@ -4363,18 +4380,21 @@ contains
43634380
if (norm_dir == 1) then
43644381
Re_shear = Re_avg_rsx_vf(j_loop, k_loop, l_loop, 1)
43654382
Re_bulk = Re_avg_rsx_vf(j_loop, k_loop, l_loop, 2)
4383+
!$acc loop seq
43664384
do i_dim = 1, num_dims
43674385
vel_src_at_interface(i_dim) = vel_src_rsx_vf(j_loop, k_loop, l_loop, i_dim)
43684386
end do
43694387
else if (norm_dir == 2) then
43704388
Re_shear = Re_avg_rsy_vf(k_loop, j_loop, l_loop, 1)
43714389
Re_bulk = Re_avg_rsy_vf(k_loop, j_loop, l_loop, 2)
4390+
!$acc loop seq
43724391
do i_dim = 1, num_dims
43734392
vel_src_at_interface(i_dim) = vel_src_rsy_vf(k_loop, j_loop, l_loop, i_dim)
43744393
end do
43754394
else
43764395
Re_shear = Re_avg_rsz_vf(l_loop, k_loop, j_loop, 1)
43774396
Re_bulk = Re_avg_rsz_vf(l_loop, k_loop, j_loop, 2)
4397+
!$acc loop seq
43784398
do i_dim = 1, num_dims
43794399
vel_src_at_interface(i_dim) = vel_src_rsz_vf(l_loop, k_loop, j_loop, i_dim)
43804400
end do
@@ -4384,6 +4404,7 @@ contains
43844404
current_tau_shear = 0.0_wp
43854405
call s_calculate_shear_stress_tensor(vel_grad_avg, Re_shear, divergence_v, current_tau_shear)
43864406

4407+
!$acc loop seq
43874408
do i_dim = 1, num_dims
43884409
flux_src_vf(momxb + i_dim - 1)%sf(j_loop, k_loop, l_loop) = &
43894410
flux_src_vf(momxb + i_dim - 1)%sf(j_loop, k_loop, l_loop) - current_tau_shear(norm_dir, i_dim)
@@ -4398,6 +4419,7 @@ contains
43984419
current_tau_bulk = 0.0_wp
43994420
call s_calculate_bulk_stress_tensor(Re_bulk, divergence_v, current_tau_bulk)
44004421

4422+
!$acc loop seq
44014423
do i_dim = 1, num_dims
44024424
flux_src_vf(momxb + i_dim - 1)%sf(j_loop, k_loop, l_loop) = &
44034425
flux_src_vf(momxb + i_dim - 1)%sf(j_loop, k_loop, l_loop) - current_tau_bulk(norm_dir, i_dim)

0 commit comments

Comments
 (0)