Add acc directives to enable executing atm_bdy_set_scalars_work on gp…

…u's.
MPAS-Dev · Jan 16, 2025 · a7d955d · a7d955d
1 parent d035210
commit a7d955d
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/src/core_atmosphere/dynamics/mpas_atm_time_integration.F b/src/core_atmosphere/dynamics/mpas_atm_time_integration.F
@@ -6913,13 +6913,18 @@ subroutine atm_bdy_set_scalars_work( scalars_driving, scalars_new, &
 
       !---
 
+      !$acc enter data copyin(scalars_driving)
+      !$acc enter data create(scalars_new)
+      !$acc parallel default(present)
+      !$acc loop gang
       do iCell = cellSolveStart, cellSolveEnd ! threaded over cells
 
          if ( bdyMaskCell(iCell) > nRelaxZone) then ! specified zone
 
             !  update the specified-zone values
             !
 !DIR$ IVDEP
+           !$acc loop vector collapse(2)
             do k=1,nVertLevels
                do iScalar = 1, num_scalars
                   scalars_new(iScalar,k,iCell) = scalars_driving(iScalar,k,iCell)
@@ -6929,6 +6934,9 @@ subroutine atm_bdy_set_scalars_work( scalars_driving, scalars_new, &
          end if
 
       end do  ! updates now in temp storage
+      !$acc end parallel
+      !$acc exit data copyout(scalars_new)
+      !$acc exit data delete(scalars_new, scalars_driving)
 
    end subroutine atm_bdy_set_scalars_work