Skip to content

Commit 31616a9

Browse files
author
Anand
committed
Format
1 parent f074f32 commit 31616a9

File tree

1 file changed

+11
-12
lines changed

1 file changed

+11
-12
lines changed

src/simulation/m_mpi_proxy.fpp

Lines changed: 11 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -345,18 +345,17 @@ contains
345345
#:endfor
346346
call nvtxEndRange ! Packbuf
347347
348-
349348
#:for rdma_mpi in [False, True]
350349
if (rdma_mpi .eqv. ${'.true.' if rdma_mpi else '.false.'}$) then
351350
#:if rdma_mpi
352351
#:call GPU_HOST_DATA(use_device='[ib_buff_send, ib_buff_recv]')
353-
354-
call nvtxStartRange("IB-MARKER-SENDRECV-RDMA")
355-
call MPI_SENDRECV( &
356-
ib_buff_send, buffer_count, MPI_INTEGER, dst_proc, send_tag, &
357-
ib_buff_recv, buffer_count, MPI_INTEGER, src_proc, recv_tag, &
358-
MPI_COMM_WORLD, MPI_STATUS_IGNORE, ierr)
359-
call nvtxEndRange
352+
353+
call nvtxStartRange("IB-MARKER-SENDRECV-RDMA")
354+
call MPI_SENDRECV( &
355+
ib_buff_send, buffer_count, MPI_INTEGER, dst_proc, send_tag, &
356+
ib_buff_recv, buffer_count, MPI_INTEGER, src_proc, recv_tag, &
357+
MPI_COMM_WORLD, MPI_STATUS_IGNORE, ierr)
358+
call nvtxEndRange
360359
361360
#:endcall GPU_HOST_DATA
362361
$:GPU_WAIT()
@@ -367,10 +366,10 @@ contains
367366
368367
call nvtxStartRange("IB-MARKER-SENDRECV-NO-RMDA")
369368
call MPI_SENDRECV( &
370-
ib_buff_send, buffer_count, MPI_INTEGER, dst_proc, send_tag, &
371-
ib_buff_recv, buffer_count, MPI_INTEGER, src_proc, recv_tag, &
372-
MPI_COMM_WORLD, MPI_STATUS_IGNORE, ierr)
373-
call nvtxEndRange
369+
ib_buff_send, buffer_count, MPI_INTEGER, dst_proc, send_tag, &
370+
ib_buff_recv, buffer_count, MPI_INTEGER, src_proc, recv_tag, &
371+
MPI_COMM_WORLD, MPI_STATUS_IGNORE, ierr)
372+
call nvtxEndRange
374373
375374
call nvtxStartRange("IB-MARKER-HOST2DEV")
376375
$:GPU_UPDATE(device='[ib_buff_recv]')

0 commit comments

Comments
 (0)