Scatter real data from root node (array of rank 2)
Type | Intent | Optional | Attributes | Name | ||
---|---|---|---|---|---|---|
real(kind=dp), | intent(inout), | dimension(:, :) | :: | array | local array for getting data |
|
integer, | intent(in) | :: | localcount | localcount elements will be fetched from the root node |
||
real(kind=dp), | intent(inout), | dimension(:, :) | :: | rootglobalarray | array on the root node from which data will be sent |
|
integer, | intent(in), | dimension(num_nodes) | :: | counts | how data should be partitioned, see MPI documentation or function comms_array_split |
|
integer, | intent(in), | dimension(num_nodes) | :: | displs |
subroutine comms_scatterv_real_2(array, localcount, rootglobalarray, counts, displs)
!! Scatter real data from root node (array of rank 2)
implicit none
real(kind=dp), dimension(:, :), intent(inout) :: array
!! local array for getting data
integer, intent(in) :: localcount
!! localcount elements will be fetched from the root node
real(kind=dp), dimension(:, :), intent(inout) :: rootglobalarray
!! array on the root node from which data will be sent
integer, dimension(num_nodes), intent(in) :: counts
!! how data should be partitioned, see MPI documentation or function comms_array_split
integer, dimension(num_nodes), intent(in) :: displs
#ifdef MPI
integer :: error
call MPI_scatterv(rootglobalarray, counts, displs, MPI_double_precision, &
array, localcount, MPI_double_precision, root_id, mpi_comm_world, error)
if (error .ne. MPI_success) then
call io_error('Error in comms_scatterv_real_2')
end if
#else
call dcopy(localcount, rootglobalarray, 1, array, 1)
#endif
return
end subroutine comms_scatterv_real_2