Skip to content

Fb reg gpuport #40

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
wants to merge 15 commits into
base: develop
Choose a base branch
from
Open

Fb reg gpuport #40

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
15 commits
Select commit Hold shift + click to select a range
178e48d
Initial regular grid GPU porting effort.
UKMO-lsampson Apr 13, 2023
35455e6
Added switches for GPU regression tests
UKMO-lsampson Apr 13, 2023
e455978
Added forth regression test switch file
UKMO-lsampson Apr 13, 2023
736b679
Updated matrix.base for regular grid propagation GPU tests
UKMO-lsampson Apr 13, 2023
29e9315
Merge branch 'NOAA-EMC:develop' into fb_reg_gpuport
ukmo-ccbunney May 2, 2023
d1ca545
Merge branch 'NOAA-EMC:develop' into fb_reg_gpuport
ukmo-ccbunney May 11, 2023
5c536c3
Merge branch 'NOAA-EMC:develop' into fb_reg_gpuport
ukmo-ccbunney Jun 16, 2023
fea89a1
Merge branch 'NOAA-EMC:develop' into fb_reg_gpuport
ukmo-ccbunney Jul 28, 2023
90b4afa
Merge branch 'NOAA-EMC:develop' into fb_reg_gpuport
ukmo-ccbunney Nov 10, 2023
9ebcd5a
Merge branch 'NOAA-EMC:develop' into fb_reg_gpuport
ukmo-ccbunney Nov 29, 2023
56fc31c
Merge remote-tracking branch 'origin/develop' into fb_reg_gpuport
ukmo-ccbunney Jan 5, 2024
d4f17e9
Merge branch 'develop' into fb_reg_gpuport
ukmo-ccbunney Apr 12, 2024
7b8e1c4
Merge branch 'develop' into fb_reg_gpuport
ukmo-ccbunney May 13, 2024
abe1a4b
Merge branch 'develop' into fb_reg_gpuport
ukmo-ccbunney Jun 18, 2024
3870fac
Merge remote-tracking branch 'origin/develop' into fb_reg_gpuport
ukmo-ccbunney Sep 4, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
54 changes: 51 additions & 3 deletions model/src/w3pro2md.F90
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -855,6 +855,12 @@ SUBROUTINE W3XYP2 ( ISP, DTG, MAPSTA, MAPFS, VQ, VGX, VGY )
WRITE (NDST,9010)
#endif
!
#ifdef W3_GPU
!$ACC DATA CREATE(VLCFLY, VLCFLX, VDXX, VDYY, VDXY, CXTOT, CYTOT) &
!$ACC CREATE(VFDIFX_FAC, VFDIFY_FAC, VFDIFC_FAC, VQ_OLD) &
!$ACC CREATE(HQFAC, HPFAC)
!$ACC KERNELS
#endif
VLCFLX = 0.
VLCFLY = 0.
VFDIFX = 0.
Expand All @@ -879,6 +885,8 @@ SUBROUTINE W3XYP2 ( ISP, DTG, MAPSTA, MAPFS, VQ, VGX, VGY )
!
#ifdef W3_OMPH
!$OMP PARALLEL DO PRIVATE (ISEA, IXY)
#elif W3_GPU
!$ACC LOOP INDEPENDENT
#endif
!
DO ISEA=1, NSEA
Expand All @@ -899,11 +907,17 @@ SUBROUTINE W3XYP2 ( ISP, DTG, MAPSTA, MAPFS, VQ, VGX, VGY )
!
#ifdef W3_OMPH
!$OMP END PARALLEL DO
#elif W3_GPU
!$ACC END KERNELS
#endif
!
IF ( FLCUR ) THEN
#ifdef W3_T
WRITE (NDST,9022)
#endif
#ifdef W3_GPU
!$ACC KERNELS
!$ACC LOOP INDEPENDENT
#endif
DO ISEA=1, NSEA
IXY = MAPSF(ISEA,3)
Expand All @@ -914,11 +928,17 @@ SUBROUTINE W3XYP2 ( ISP, DTG, MAPSTA, MAPFS, VQ, VGX, VGY )
VQ(IXY), CXTOT(IXY), CYTOT(IXY)
#endif
END DO
#ifdef W3_GPU
!$ACC END KERNELS
#endif
END IF

!
#ifdef W3_OMPH
!$OMP PARALLEL DO PRIVATE (ISEA, IX, IY, IXY, CP, CQ)
#elif W3_GPU
!$ACC KERNELS
!$ACC LOOP INDEPENDENT
#endif
!
DO ISEA=1, NSEA
Expand All @@ -933,6 +953,8 @@ SUBROUTINE W3XYP2 ( ISP, DTG, MAPSTA, MAPFS, VQ, VGX, VGY )
!
#ifdef W3_OMPH
!$OMP END PARALLEL DO
#elif W3_GPU
!$ACC END KERNELS
#endif
!
! 2.b Diffusion coefficients
Expand All @@ -942,6 +964,9 @@ SUBROUTINE W3XYP2 ( ISP, DTG, MAPSTA, MAPFS, VQ, VGX, VGY )
#ifdef W3_OMPH
!$OMP PARALLEL DO PRIVATE (ISEA, IX, IY, IXY, &
!$OMP& DCELL, XWIND, TFAC, DSS, DNN)
#elif W3_GPU
!$ACC KERNELS
!$ACC LOOP INDEPENDENT
#endif
!
DO ISEA=1, NSEA
Expand Down Expand Up @@ -978,6 +1003,8 @@ SUBROUTINE W3XYP2 ( ISP, DTG, MAPSTA, MAPFS, VQ, VGX, VGY )
!
#ifdef W3_OMPH
!$OMP END PARALLEL DO
#elif W3_GPU
!$ACC END KERNELS
#endif
!
END IF
Expand All @@ -990,8 +1017,10 @@ SUBROUTINE W3XYP2 ( ISP, DTG, MAPSTA, MAPFS, VQ, VGX, VGY )
!
#ifdef W3_OMPH
!$OMP PARALLEL DO PRIVATE (ISEA, IX, IY, IXY )
#elif W3_GPU
!$ACC KERNELS
!$ACC LOOP INDEPENDENT
#endif
!
DO ISEA=1, NSEA
IX = MAPSF(ISEA,1)
IY = MAPSF(ISEA,2)
Expand All @@ -1001,6 +1030,8 @@ SUBROUTINE W3XYP2 ( ISP, DTG, MAPSTA, MAPFS, VQ, VGX, VGY )
!
#ifdef W3_OMPH
!$OMP END PARALLEL DO
#elif W3_GPU
!$ACC END KERNELS
#endif
!
IF ( YFIRST ) THEN
Expand Down Expand Up @@ -1055,6 +1086,9 @@ SUBROUTINE W3XYP2 ( ISP, DTG, MAPSTA, MAPFS, VQ, VGX, VGY )
!
#ifdef W3_OMPH
!$OMP PARALLEL DO PRIVATE (ISEA, IX, IY, IXY )
#elif W3_GPU
!$ACC KERNELS
!$ACC LOOP INDEPENDENT
#endif
!
DO ISEA=1, NSEA
Expand Down Expand Up @@ -1088,11 +1122,16 @@ SUBROUTINE W3XYP2 ( ISP, DTG, MAPSTA, MAPFS, VQ, VGX, VGY )
/ CG(IK,ISEA) * CLATS(ISEA)
END DO
END IF
#ifdef W3_GPU
!$ACC END KERNELS
#endif
!
! 3.c Diffusion correction
!
IF ( DTME .NE. 0. ) THEN

#ifdef W3_GPU
!$ACC KERNELS
#endif
IF ( GLOBAL ) THEN
DO IY=1, NY
VQ(IY+NX*NY) = VQ(IY)
Expand Down Expand Up @@ -1129,8 +1168,9 @@ SUBROUTINE W3XYP2 ( ISP, DTG, MAPSTA, MAPFS, VQ, VGX, VGY )
#ifdef W3_OMPH
!$OMP PARALLEL DO PRIVATE (ISEA, IX, IY, IXY, &
!$OMP& QXX, QYY, QXY, DVQ )
#elif W3_GPU
!$ACC LOOP INDEPENDENT
#endif
!
DO IP=1, NACT
IXY = MAPAXY(IP)
ISEA = MAPFS(IXY)
Expand Down Expand Up @@ -1200,6 +1240,8 @@ SUBROUTINE W3XYP2 ( ISP, DTG, MAPSTA, MAPFS, VQ, VGX, VGY )
!
#ifdef W3_OMPH
!$OMP END PARALLEL DO
#elif W3_GPU
!$ACC END KERNELS
#endif
!
END IF
Expand All @@ -1215,6 +1257,9 @@ SUBROUTINE W3XYP2 ( ISP, DTG, MAPSTA, MAPFS, VQ, VGX, VGY )
!
#ifdef W3_OMPH
!$OMP PARALLEL DO PRIVATE (ISEA, IXY )
#elif W3_GPU
!$ACC KERNELS
!$ACC LOOP INDEPENDENT
#endif
!
DO ISEA=1, NSEA
Expand All @@ -1231,6 +1276,9 @@ SUBROUTINE W3XYP2 ( ISP, DTG, MAPSTA, MAPFS, VQ, VGX, VGY )
!
#ifdef W3_OMPH
!$OMP END PARALLEL DO
#elif W3_GPU
!$ACC END KERNELS
!$ACC END DATA
#endif
!
RETURN
Expand Down
17 changes: 17 additions & 0 deletions model/src/w3uno2md.F90
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -983,6 +983,10 @@ SUBROUTINE W3UNO2s (MX, MY, NX, NY, CFLL, TRANS, Q, BCLOSE, &
!
! 1. Initialize aux. array FLA and closure ------------------------- *
!
#ifdef W3_GPU
!$ACC DATA CREATE(FLA)
!$ACC KERNELS
#endif
FLA = 0.
!
IF ( BCLOSE ) THEN
Expand Down Expand Up @@ -1026,6 +1030,8 @@ SUBROUTINE W3UNO2s (MX, MY, NX, NY, CFLL, TRANS, Q, BCLOSE, &
!$OMP QBO, IX, IY, IY2, IX2, QN &
#endif
!$OMP IXYC, IXYD, QB)
#elif W3_GPU
!$ACC LOOP INDEPENDENT
#endif
!
DO IP=1, NB0
Expand Down Expand Up @@ -1073,6 +1079,9 @@ SUBROUTINE W3UNO2s (MX, MY, NX, NY, CFLL, TRANS, Q, BCLOSE, &
WRITE (NDST,9011) NB1-NB0, 'BOUNDARY ABOVE'
#endif
!
#ifdef W3_GPU
!$ACC LOOP INDEPENDENT
#endif
DO IP=NB0+1, NB1
IXY = MAPBOU(IP)
CFL = CFLL(IXY)
Expand All @@ -1099,6 +1108,9 @@ SUBROUTINE W3UNO2s (MX, MY, NX, NY, CFLL, TRANS, Q, BCLOSE, &
WRITE (NDST,9011) NB2-NB1, 'BOUNDARY BELOW'
#endif
!
#ifdef W3_GPU
!$ACC LOOP INDEPENDENT
#endif
DO IP=NB1+1, NB2
IXY = MAPBOU(IP)
CFL = CFLL(IXY+INC)
Expand Down Expand Up @@ -1141,6 +1153,8 @@ SUBROUTINE W3UNO2s (MX, MY, NX, NY, CFLL, TRANS, Q, BCLOSE, &
!$OMP PRIVATE(QOLD), &
#endif
!$OMP PRIVATE (IP, IXY, JN, JP)
#elif W3_GPU
!$ACC LOOP INDEPENDENT
#endif
!
DO IP=1, NACT
Expand Down Expand Up @@ -1172,6 +1186,9 @@ SUBROUTINE W3UNO2s (MX, MY, NX, NY, CFLL, TRANS, Q, BCLOSE, &
!
#ifdef W3_OMPH
!$OMP END PARALLEL DO
#elif W3_GPU
!$ACC END KERNELS
!$ACC END DATA
#endif
!
#ifdef W3_T0
Expand Down
6 changes: 6 additions & 0 deletions model/src/w3wavemd.F90
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -1853,6 +1853,9 @@ SUBROUTINE W3WAVE ( IMOD, ODAT, TEND, STAMP, NO_OUT &
!
!
! Initialize FIELD variable
#ifdef W3_GPU
!$ACC UPDATE DEVICE(VA)
#endif
FIELD = 0.
!
DO ISPEC=1, NSPEC
Expand Down Expand Up @@ -1919,6 +1922,9 @@ SUBROUTINE W3WAVE ( IMOD, ODAT, TEND, STAMP, NO_OUT &

END IF
END DO
#ifdef W3_GPU
!$ACC UPDATE SELF(VA)
#endif
!
#ifdef W3_MPI
IF ( NRQSG1 .GT. 0 ) THEN
Expand Down
8 changes: 8 additions & 0 deletions regtests/bin/matrix.base
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -263,6 +263,10 @@
echo "$rtst -s PR2_UNO -w work_PR2_UNO $ww3 ww3_tp2.2" >> matrix.body
echo "$rtst -s PR2_UNO -w work_PR2_UNO $ww3 ww3_tp2.3" >> matrix.body
echo "$rtst -s PR2_UNO -w work_PR2_UNO $ww3 ww3_tp2.4" >> matrix.body
echo "$rtst -s PR2_UNO_GPU -w work_PR2_UNO_GPU $ww3 ww3_tp2.1" >> matrix.body
echo "$rtst -s PR2_UNO_GPU -w work_PR2_UNO_GPU $ww3 ww3_tp2.2" >> matrix.body
echo "$rtst -s PR2_UNO_GPU -w work_PR2_UNO_GPU $ww3 ww3_tp2.3" >> matrix.body
echo "$rtst -s PR2_UNO_GPU -w work_PR2_UNO_GPU $ww3 ww3_tp2.4" >> matrix.body
echo "$rtst -s PR2_UNO -w work_PR2_UNO_curv -g curv $ww3 ww3_tp2.4" >> matrix.body
echo "$rtst -s PR2_UNO -w work_PR2_UNO $ww3 ww3_tp2.5" >> matrix.body
echo "$rtst -s PR2_UNO -w work_PR2_UNO_a -g a $ww3 ww3_tp2.9" >> matrix.body
Expand Down Expand Up @@ -312,6 +316,10 @@
echo "$rtst -s PR2_UNO_MPI -w work_PR2_UNO_MPI -f -p $mpi -n $np $ww3 ww3_tp2.2" >> matrix.body
echo "$rtst -s PR2_UNO_MPI -w work_PR2_UNO_MPI -f -p $mpi -n $np $ww3 ww3_tp2.3" >> matrix.body
echo "$rtst -s PR2_UNO_MPI -w work_PR2_UNO_MPI -f -p $mpi -n $np $ww3 ww3_tp2.4" >> matrix.body
echo "$rtst -s PR2_UNO_MPI_GPU -w work_PR2_UNO_MPI_GPU -f -p $mpi -n $np $ww3 ww3_tp2.1" >> matrix.body
echo "$rtst -s PR2_UNO_MPI_GPU -w work_PR2_UNO_MPI_GPU -f -p $mpi -n $np $ww3 ww3_tp2.2" >> matrix.body
echo "$rtst -s PR2_UNO_MPI_GPU -w work_PR2_UNO_MPI_GPU -f -p $mpi -n $np $ww3 ww3_tp2.3" >> matrix.body
echo "$rtst -s PR2_UNO_MPI_GPU -w work_PR2_UNO_MPI_GPU -f -p $mpi -n $np $ww3 ww3_tp2.4" >> matrix.body
echo "$rtst -s PR2_UNO_MPI -w work_PR2_UNO_curv_MPI -g curv -f -p $mpi -n $np $ww3 ww3_tp2.4" >> matrix.body
echo "$rtst -s PR2_UNO_MPI -w work_PR2_UNO_MPI -f -p $mpi -n $np $ww3 ww3_tp2.5" >> matrix.body
echo "$rtst -s PR2_UNO_MPI -w work_PR2_UNO_a_MPI -g a -f -p $mpi -n $np $ww3 ww3_tp2.9" >> matrix.body
Expand Down
1 change: 1 addition & 0 deletions regtests/ww3_tp2.1/input/switch_PR2_UNO_GPU
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
NOGRB SHRD GPU PR2 UNO FLX2 LN0 ST0 NL0 BT0 DB0 TR0 BS0 IC0 IS0 REF0 WNT1 WNX1 CRT1 CRX1 O0 O1 O2 O3 O4 O5 O6 O7 O10 O11
1 change: 1 addition & 0 deletions regtests/ww3_tp2.1/input/switch_PR2_UNO_MPI_GPU
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
NOGRB DIST MPI GPU PR2 UNO FLX2 LN0 ST0 NL0 BT0 DB0 TR0 BS0 IC0 IS0 REF0 WNT1 WNX1 CRT1 CRX1 O0 O1 O2 O3 O4 O5 O6 O7 O10 O11
1 change: 1 addition & 0 deletions regtests/ww3_tp2.2/input/switch_PR2_UNO_GPU
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
NOGRB SHRD PR2 UNO FLX2 LN0 ST0 NL0 BT0 DB0 TR0 BS0 IC0 IS0 REF0 WNT1 WNX1 CRT1 CRX1 O0 O1 O2 O3 O4 O5 O6 O7 O10 O11
1 change: 1 addition & 0 deletions regtests/ww3_tp2.2/input/switch_PR2_UNO_MPI_GPU
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
NOGRB DIST MPI GPU PR2 UNO FLX2 LN0 ST0 NL0 BT0 DB0 TR0 BS0 IC0 IS0 REF0 WNT1 WNX1 CRT1 CRX1 O0 O1 O2 O3 O4 O5 O6 O7 O10 O11
1 change: 1 addition & 0 deletions regtests/ww3_tp2.3/input/switch_PR2_UNO_GPU
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
NOGRB SHRD GPU PR2 UNO FLX2 LN0 ST0 NL0 BT0 DB0 TR0 BS0 IC0 IS0 REF0 WNT1 WNX1 CRT1 CRX1 O0 O1 O2 O3 O4 O5 O6 O7 O10 O11
1 change: 1 addition & 0 deletions regtests/ww3_tp2.3/input/switch_PR2_UNO_MPI_GPU
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
NOGRB DIST MPI GPU PR2 UNO FLX2 LN0 ST0 NL0 BT0 DB0 TR0 BS0 IC0 IS0 REF0 WNT1 WNX1 CRT1 CRX1 O0 O1 O2 O3 O4 O5 O6 O7 O10 O11
1 change: 1 addition & 0 deletions regtests/ww3_tp2.4/input/switch_PR2_UNO_GPU
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
NOGRB SHRD GPU PR2 UNO FLX2 LN0 ST0 NL0 BT0 DB0 TR0 BS0 IC0 IS0 REF0 WNT1 WNX1 CRT1 CRX1 O0 O1 O2 O3 O4 O5 O6 O7 O10 O11
1 change: 1 addition & 0 deletions regtests/ww3_tp2.4/input/switch_PR2_UNO_MPI_GPU
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
NOGRB DIST MPI GPU PR2 UNO FLX2 LN0 ST0 NL0 BT0 DB0 TR0 BS0 IC0 IS0 REF0 WNT1 WNX1 CRT1 CRX1 O0 O1 O2 O3 O4 O5 O6 O7 O10 O11
Loading