@@ -444,7 +444,39 @@ contains
444444 call s_update_lagrange_tdv_rk(stage= 1 )
445445 end if
446446
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
448480
449481 if (bodyForces) call s_apply_bodyforces(q_cons_ts(1 )%vf, q_prim_vf, rhs_vf, dt)
450482
0 commit comments