Skip to content

Commit

Permalink
Fix for v large data transfers
Browse files Browse the repository at this point in the history
  • Loading branch information
awnawab committed Jan 30, 2024
1 parent 9b318a1 commit 5ef89da
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 3 deletions.
5 changes: 3 additions & 2 deletions field_RANKSUFF_data_module.fypp
Original file line number Diff line number Diff line change
Expand Up @@ -110,11 +110,12 @@ CONTAINS
#ifdef _OPENACC
USE OPENACC
#endif
USE, INTRINSIC :: ISO_FORTRAN_ENV, ONLY : INT64
${ft.type}$, POINTER :: HST (${ft.shape}$), DEV (${ft.shape}$)
LOGICAL, INTENT (IN) :: MAP_DEVPTR
INTEGER (KIND=JPIM), INTENT (IN) :: KDIR
INTEGER (KIND=JPIM), OPTIONAL, INTENT (IN) :: QUEUE
INTEGER (KIND=JPIM) :: ISIZE
INTEGER (KIND=INT64) :: ISIZE
INTEGER :: ${', '.join (['J'] + list (map (lambda i: 'J' + str (i+1), range (d, ft.rank))))}$
#ifdef _OPENACC
TYPE(C_DEVPTR) :: DEVPTR
Expand All @@ -139,7 +140,7 @@ CONTAINS
#:if d == 0
${indent}$ ISIZE = KIND (HST)
#:else
${indent}$ ISIZE = KIND (HST) * SIZE (HST (${ar}$))
${indent}$ ISIZE = SIZEOF (HST)
#:endif
${indent}$ IF (KDIR == NH2D) THEN
#ifdef _OPENACC
Expand Down
2 changes: 1 addition & 1 deletion field_basic_module.F90
Original file line number Diff line number Diff line change
Expand Up @@ -35,7 +35,7 @@ MODULE FIELD_BASIC_MODULE
TYPE(GPU_STATS) :: STATS

LOGICAL :: LOBJECT_COPIED = .FALSE.
LOGICAL :: MAP_DEVPTR = .TRUE.
LOGICAL :: MAP_DEVPTR = .FALSE.

CONTAINS
PROCEDURE (FIELD_BASIC_SYNC), DEFERRED :: SYNC_HOST_RDWR
Expand Down

0 comments on commit 5ef89da

Please sign in to comment.