Homec4science
Diffusion COSOlver (new_coulomb) Browse

cosolver/src/basic_mpi_mod.f90
83b8f38f6c51new_coulomb

Last Change
Raw File

basic_mpi_mod.f90
View Options

module basic_mpi
! Basic MPI module
use prec_const
IMPLICIT NONE
PUBLIC
! MPI handles and variables
INTEGER :: ierr ! flag for MPI error
INTEGER :: nprocs ! number of processors
INTEGER :: me =0 ! Rank of processor
INTEGER :: meCe =0 ! Electron Collision Processor rank
INTEGER :: meCi =0 ! Ion Collision Processor rank
INTEGER :: meCaa =0 ! Self Collision Processor rank
INTEGER :: meCeiT = 0
INTEGER :: meCeiF = 0
INTEGER :: meCieT = 0
INTEGER :: meCieF = 0
INTEGER :: meCee = 0
INTEGER :: meCii =0
CHARACTER(len=3) :: me_str
! number of MPI proc in the enegry component
INTEGER :: Nprocs_rows_ii
! GK ii Cartesian topology
INTEGER :: mycoords_ii(2)
! MPI GROUPS
INTEGER, DIMENSION(:),ALLOCATABLE :: ranks_e,ranks_i,ranks_self
INTEGER :: grp_e,grp_i,grp_self
INTEGER :: grp_world
! Group comm
INTEGER :: comm_e,comm_i,comm_self
! Cartesian comm
INTEGER :: comm_cart_e,comm_cart_i,comm_cart_self
! GRROUP RANK
INTEGER :: me_e,me_i,me_self
LOGICAL :: is_e = .false.
LOGICAL :: is_i = .false.
LOGICAL :: is_self = .false.
!
! number of mpi task per type of collisions
INTEGER :: nprocs_ei = 1
INTEGER :: nprocs_ie = 1
INTEGER :: nprocs_self = 1
! IO units
INTEGER :: lu_in ! File duplicated from STDIN
INTEGER :: lu_job = 91 ! myjob file
INTEGER :: lu_out = 11 ! Collisional moment output
integer, parameter :: nscr = 6 ! standard ouput
INTEGER :: nkbd = 5 ! standard input
character(2) :: strlu_
contains
subroutine parallelization_readinputs
! read the parallelization inputs
use prec_const
implicit none
!
namelist /parallelization/ nprocs_ei, nprocs_ie, nprocs_self
IF(me .eq. 0) THEN
READ(lu_in,parallelization)
!! write(*,parallelization)
ENDIF
CALL MPI_bcast(nprocs_ei,1,MPI_INTEGER,0,MPI_COMM_WORLD,ierr)
CALL MPI_bcast(nprocs_ie,1,MPI_INTEGER,0,MPI_COMM_WORLD,ierr)
CALL MPI_bcast(nprocs_self,1,MPI_INTEGER,0,MPI_COMM_WORLD,ierr)
!
IF( me .eq. 0) WRITE(*,*) 'Number of MPI tasks for ei(T-F)-colls: ', nprocs_ei
IF( me .eq. 0) WRITE(*,*) 'Number of MPI tasks for ie(T-F)-colls: ', nprocs_ie
IF( me .eq. 0) WRITE(*,*) 'Number of MPI tasks for self-colls: ', nprocs_self
IF (( nprocs_ei + nprocs_ie + nprocs_self .ne. nprocs ) .and. ( nprocs .gt. 1 )) then
IF ( me .eq. 0) print*, "Number of MPI task requested does not match the number allocated ... "
CALL MPI_Abort(MPI_COMM_WORLD,-2,ierr)
ENDIF
!
end subroutine parallelization_readinputs
end module basic_mpi
c4science · Help