@@ -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