forked from UK-MAC/CloverLeaf_OpenACC_3D
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathflux_calc_kernel.f90
106 lines (93 loc) · 4.05 KB
/
flux_calc_kernel.f90
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
!Crown Copyright 2012 AWE.
!
! This file is part of CloverLeaf.
!
! CloverLeaf is free software: you can redistribute it and/or modify it under
! the terms of the GNU General Public License as published by the
! Free Software Foundation, either version 3 of the License, or (at your option)
! any later version.
!
! CloverLeaf is distributed in the hope that it will be useful, but
! WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
! FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for more
! details.
!
! You should have received a copy of the GNU General Public License along with
! CloverLeaf. If not, see http://www.gnu.org/licenses/.
!> @brief Fortran flux kernel.
!> @author Wayne Gaudin
!> @details The edge volume fluxes are calculated based on the velocity fields.
MODULE flux_calc_kernel_module
CONTAINS
SUBROUTINE flux_calc_kernel(x_min,x_max,y_min,y_max,z_min,z_max,dt, &
xarea, &
yarea, &
zarea, &
xvel0, &
yvel0, &
zvel0, &
xvel1, &
yvel1, &
zvel1, &
vol_flux_x, &
vol_flux_y, &
vol_flux_z )
IMPLICIT NONE
INTEGER :: x_min, x_max, y_min, y_max, z_min, z_max
REAL(KIND=8) :: dt
REAL(KIND=8), DIMENSION(x_min-2:x_max+3,y_min-2:y_max+2,z_min-2:z_max+2) :: xarea
REAL(KIND=8), DIMENSION(x_min-2:x_max+2,y_min-2:y_max+3,z_min-2:z_max+2) :: yarea
REAL(KIND=8), DIMENSION(x_min-2:x_max+2,y_min-2:y_max+2,z_min-2:z_max+3) :: zarea
REAL(KIND=8), DIMENSION(x_min-2:x_max+3,y_min-2:y_max+3,z_min-2:z_max+3) :: xvel0,yvel0,zvel0
REAL(KIND=8), DIMENSION(x_min-2:x_max+3,y_min-2:y_max+3,z_min-2:z_max+3) :: xvel1,yvel1,zvel1
REAL(KIND=8), DIMENSION(x_min-2:x_max+3,y_min-2:y_max+2,z_min-2:z_max+2) :: vol_flux_x
REAL(KIND=8), DIMENSION(x_min-2:x_max+2,y_min-2:y_max+3,z_min-2:z_max+2) :: vol_flux_y
REAL(KIND=8), DIMENSION(x_min-2:x_max+2,y_min-2:y_max+2,z_min-2:z_max+3) :: vol_flux_z
INTEGER :: j,k,l
!$ACC DATA &
!$ACC PCOPY(xvel0,yvel0,zvel0,xvel1,yvel1,zvel1,xarea,yarea,zarea,vol_flux_x,vol_flux_y,vol_flux_z)
!$ACC KERNELS
!$ACC LOOP INDEPENDENT
DO l=z_min,z_max
!$ACC LOOP INDEPENDENT
DO k=y_min,y_max
!$ACC LOOP INDEPENDENT
DO j=x_min,x_max+1
vol_flux_x(j,k,l)=0.125_8*dt*xarea(j,k,l) &
*(xvel0(j,k,l)+xvel0(j,k+1,l)+xvel0(j,k,l+1)+xvel0(j,k+1,l+1) &
+xvel1(j,k,l)+xvel1(j,k+1,l)+xvel1(j,k,l+1)+xvel1(j,k+1,l+1))
ENDDO
ENDDO
ENDDO
!$ACC END KERNELS
!$ACC KERNELS
!$ACC LOOP INDEPENDENT
DO l=z_min,z_max
!$ACC LOOP INDEPENDENT
DO k=y_min,y_max+1
!$ACC LOOP INDEPENDENT
DO j=x_min,x_max
vol_flux_y(j,k,l)=0.125_8*dt*yarea(j,k,l) &
*(yvel0(j,k,l)+yvel0(j+1,k,l)+yvel0(j,k,l+1)+yvel0(j+1,k,l+1) &
+yvel1(j,k,l)+yvel1(j+1,k,l)+yvel1(j,k,l+1)+yvel1(j+1,k,l+1))
ENDDO
ENDDO
ENDDO
!$ACC END KERNELS
!$ACC KERNELS
!$ACC LOOP INDEPENDENT
DO l=z_min,z_max+1
!$ACC LOOP INDEPENDENT
DO k=y_min,y_max
!$ACC LOOP INDEPENDENT
DO j=x_min,x_max
vol_flux_z(j,k,l)=0.125_8*dt*zarea(j,k,l) &
*(zvel0(j,k,l)+zvel0(j+1,k,l)+zvel0(j+1,k,l)+zvel0(j+1,k+1,l) &
+zvel1(j,k,l)+zvel1(j+1,k,l)+zvel1(j,k+1,l)+zvel1(j+1,k+1,l))
ENDDO
ENDDO
ENDDO
!$ACC END KERNELS
!$ACC END DATA
END SUBROUTINE flux_calc_kernel
END MODULE flux_calc_kernel_module