Changeset 1000 for codes/icosagcm/trunk/src

Timestamp:

01/10/20 12:20:51 (4 years ago)

Author:

adurocher

Message:

transfert_mpi : fixed tmp array copy issue on GPU

PGI doesn't give the GPU version (host_data use device) of 'message_*(i)%buff' when calling MPI_send_init.
Using 'message_*(i)%buff(1)' instead solves the problem

File:

• codes/icosagcm/trunk/src/parallel/transfert_mpi.f90 (modified) (1 diff)

codes/icosagcm/trunk/src/parallel/transfert_mpi.f90

@@ -336 +336 @@
       call MPI_Request_free(message%mpi_requests_in(i), ierr)
       !$acc host_data use_device(message%mpi_buffer_out(i)%buff)
-      call MPI_Send_Init( message%mpi_buffer_out(i)%buff, message%mpi_buffer_out(i)%n, MPI_REAL8, i,&
+      ! /!\ buff(1) is important for PGI to avoid temporary array copy
+      call MPI_Send_Init( message%mpi_buffer_out(i)%buff(1), message%mpi_buffer_out(i)%n, MPI_REAL8, i,&
                           0, comm_icosa, message%mpi_requests_out(i), ierr )
       !$acc end host_data
       !$acc host_data use_device(message%mpi_buffer_in(i)%buff)
-      call MPI_Recv_Init( message%mpi_buffer_in(i)%buff, message%mpi_buffer_in(i)%n, MPI_REAL8, i,&
+      call MPI_Recv_Init( message%mpi_buffer_in(i)%buff(1), message%mpi_buffer_in(i)%n, MPI_REAL8, i,&
                           0, comm_icosa, message%mpi_requests_in(i), ierr )
       !$acc end host_data