Skip to content

Commit 26a3b6f

Browse files
committed
Updated allocate and deallocate macros
1 parent f74dd1d commit 26a3b6f

File tree

1 file changed

+17
-15
lines changed

1 file changed

+17
-15
lines changed

src/common/include/macros.fpp

Lines changed: 17 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -14,14 +14,16 @@
1414

1515
#:def ALLOCATE(*args)
1616
@:LOG({'@:ALLOCATE(${re.sub(' +', ' ', ', '.join(args))}$)'})
17-
allocate (${', '.join(args)}$)
18-
!$acc enter data create(${', '.join(args)}$)
17+
#:set allocated_variables = ', '.join(args)
18+
allocate (${allocated_variables}$)
19+
$:GPU_ENTER_DATA(create=('[' + allocated_variables + ']'))
1920
#:enddef ALLOCATE
2021

2122
#:def DEALLOCATE(*args)
2223
@:LOG({'@:DEALLOCATE(${re.sub(' +', ' ', ', '.join(args))}$)'})
23-
!$acc exit data delete(${', '.join(args)}$)
24-
deallocate (${', '.join(args)}$)
24+
#:set allocated_variables = ', '.join(args)
25+
$:GPU_EXIT_DATA(delete=('[' + allocated_variables + ']'))
26+
deallocate (${allocated_variables}$)
2527
#:enddef DEALLOCATE
2628

2729
#:def ACC_SETUP_VFs(*args)
@@ -32,13 +34,13 @@
3234
@:LOG({'@:ACC_SETUP_VFs(${', '.join(args)}$)'})
3335

3436
#:for arg in args
35-
!$acc enter data copyin(${arg}$)
36-
!$acc enter data copyin(${arg}$%vf)
37+
$:GPU_ENTER_DATA(copyin=('[' + arg + ']'))
38+
$:GPU_ENTER_DATA(copyin=('[' + arg + '%vf]'))
3739
if (allocated(${arg}$%vf)) then
3840
do macros_setup_vfs_i = lbound(${arg}$%vf, 1), ubound(${arg}$%vf, 1)
3941
if (associated(${arg}$%vf(macros_setup_vfs_i)%sf)) then
40-
!$acc enter data copyin(${arg}$%vf(macros_setup_vfs_i))
41-
!$acc enter data create(${arg}$%vf(macros_setup_vfs_i)%sf)
42+
$:GPU_ENTER_DATA(copyin=('[' + arg + '%vf(macros_setup_vfs_i)]'))
43+
$:GPU_ENTER_DATA(copyin=('[' + arg + '%vf(macros_setup_vfs_i)%sf]'))
4244
end if
4345
end do
4446
end if
@@ -54,9 +56,9 @@
5456
@:LOG({'@:ACC_SETUP_SFs(${', '.join(args)}$)'})
5557

5658
#:for arg in args
57-
!$acc enter data copyin(${arg}$)
59+
$:GPU_ENTER_DATA(copyin=('[' + arg + ']'))
5860
if (associated(${arg}$%sf)) then
59-
!$acc enter data create(${arg}$%sf)
61+
$:GPU_ENTER_DATA(copyin=('[' + arg + '%sf]'))
6062
end if
6163
#:endfor
6264
end block
@@ -70,18 +72,18 @@
7072
@:LOG({'@:ACC_SETUP_source_spatials(${', '.join(args)}$)'})
7173

7274
#:for arg in args
73-
!$acc enter data copyin(${arg}$)
75+
$:GPU_ENTER_DATA(copyin=('[' + arg + ']'))
7476
if (allocated(${arg}$%coord)) then
75-
!$acc enter data create(${arg}$%coord)
77+
$:GPU_ENTER_DATA(copyin=('[' + arg + '%coord]'))
7678
end if
7779
if (allocated(${arg}$%val)) then
78-
!$acc enter data create(${arg}$%val)
80+
$:GPU_ENTER_DATA(copyin=('[' + arg + '%val]'))
7981
end if
8082
if (allocated(${arg}$%angle)) then
81-
!$acc enter data create(${arg}$%angle)
83+
$:GPU_ENTER_DATA(copyin=('[' + arg + '%angle]'))
8284
end if
8385
if (allocated(${arg}$%xyz_to_r_ratios)) then
84-
!$acc enter data create(${arg}$%xyz_to_r_ratios)
86+
$:GPU_ENTER_DATA(copyin=('[' + arg + '%xyz_to_r_ratios]'))
8587
end if
8688
#:endfor
8789
end block

0 commit comments

Comments
 (0)