@@ -444,7 +444,39 @@ contains
444
444
call s_update_lagrange_tdv_rk(stage= 1 )
445
445
end if
446
446
447
- call s_evolve_q_pb_mv(1 , 1.0_wp , 0.0_wp , 1.0_wp )
447
+ !$acc parallel loop collapse(4 ) gang vector default(present)
448
+ do i = 1 , sys_size
449
+ do l = 0 , p
450
+ do k = 0 , n
451
+ do j = 0 , m
452
+ q_cons_ts(1 )%vf(i)%sf(j, k, l) = &
453
+ q_cons_ts(1 )%vf(i)%sf(j, k, l) &
454
+ + dt* rhs_vf(i)%sf(j, k, l)
455
+ end do
456
+ end do
457
+ end do
458
+ end do
459
+
460
+ !Evolve pb and mv for non- polytropic qbmm
461
+ if (qbmm .and. (.not. polytropic)) then
462
+ !$acc parallel loop collapse(5 ) gang vector default(present)
463
+ do i = 1 , nb
464
+ do l = 0 , p
465
+ do k = 0 , n
466
+ do j = 0 , m
467
+ do q = 1 , nnode
468
+ pb_ts(1 )%sf(j, k, l, q, i) = &
469
+ pb_ts(1 )%sf(j, k, l, q, i) &
470
+ + dt* rhs_pb(j, k, l, q, i)
471
+ mv_ts(1 )%sf(j, k, l, q, i) = &
472
+ mv_ts(1 )%sf(j, k, l, q, i) &
473
+ + dt* rhs_mv(j, k, l, q, i)
474
+ end do
475
+ end do
476
+ end do
477
+ end do
478
+ end do
479
+ end if
448
480
449
481
if (bodyForces) call s_apply_bodyforces(q_cons_ts(1 )%vf, q_prim_vf, rhs_vf, dt)
450
482
0 commit comments