getImageCountMPI.F90

Go to the documentation of this file.

 
program xgetImageCountMPI
    use mpi !mpi_f08, only : mpi_initialized, mpi_comm_world, mpi_comm_size, mpi_init
    implicit none
    logical :: isinit
    integer :: ierrMPI
    integer :: imageRank = -1
    integer :: imageCount = 0
    imageCountMPI_block: block
        call mpi_initialized(isinit, ierrMPI)
        if (ierrMPI /= 0) exit imageCountMPI_block
        if (.not. isinit) then
            call mpi_init(ierrMPI)
            if (ierrMPI /= 0) exit imageCountMPI_block
        end if
        call mpi_comm_size(mpi_comm_world, imageCount, ierrMPI)
        if (ierrMPI /= 0) imageCount = 0
        call mpi_comm_rank(mpi_comm_world, imageRank, ierrMPI)
        if (ierrMPI /= 0) imageRank = -1
        imageRank = imageRank + 1
    end block imageCountMPI_block
    ! All images report their rank and the MPI image count.
    ! The current bash process is responsible for capturing it.
    write(*, "(*(g0))") "imageCount=", imageCount, ". imageRank=", imageRank, "."
    call mpi_finalize(ierrMPI)
end program xgetImageCountMPI