Homec4science
Diffusion COSOlver (new_coulomb) Browse

cosolver/gkcoulomb_suite/CurlyNpjmn/src/ppsetup.f90
8cb7babca835new_coulomb

Last Change
Raw File

ppsetup.f90
View Options

SUBROUTINE PPSETUP
!
! sets up the MPI topology and distribute the work betwen the nprocs
!
use prec_const
use basic
use array
!
IMPLICIT NONE
!
INTEGER :: iproc =0
INTEGER, DIMENSION(nprocs) :: procslabels
INTEGER :: ENDOF, NN, iNN_l,EXT,GMAXX,HMAXX
INTEGER :: IT1,IT2,IT3,IT4,IT5
INTEGER,SAVE :: fid
!
IF(nprocs > 1 ) THEN
!
! computes chksz per MPI process
chksz = NbOfCoeffs/nprocs
idx_s_l = me*chksz +1
idx_e_l = idx_s_l + chksz-1
!
IF(me .eq. nprocs -1) THEN
idx_e_l = NbOfCoeffs
ENDIF
! debug
print*,'MPI 1D Top. (me/local idx):', me, idx_s_l,idx_e_l
! number of NN coeff per MPI procs
nb_l = idx_e_l - idx_s_L +1
! Allocate local Idx NN coeffs
ALLOCATE(idxCoeffs(1:nb_l,4))
idxCoeffs = 0
!
! Get idx from *.csv file
!
OPEN(fid,FILE='indices.in')
!
ENDOF = 0
NN = 1
iNN_l = 1
EXT = 0
!
DO WHILE (ENDOF .eq. 0 .and. EXT .eq. 0)
READ(fid,*,IOSTAT=ENDOF) IT1,IT2,IT3,IT4
IF( NN .ge. idx_s_l .and. NN .le. idx_e_l) THEN
idxCoeffs(iNN_l,1) = IT1
idxCoeffs(iNN_l,2) = IT2
idxCoeffs(iNN_l,3) = IT3
idxCoeffs(iNN_l,4) = IT4
iNN_l = iNN_l +1
ENDIF
NN = NN+1
ENDDO
!
CLOSE(fid)
! !
! Get local maximum and minimum indices
pmin_l = minval(idxCoeffs(:,1))
pmax_l = maxval(idxCoeffs(:,1))
!
jmin_l = minval(idxCoeffs(:,2))
jmax_l = maxval(idxCoeffs(:,2))
!
mmin_l = minval(idxCoeffs(:,3))
mmax_l = maxval(idxCoeffs(:,3))
!
nmin_l = minval(idxCoeffs(:,4))
nmax_l = maxval(idxCoeffs(:,4))
!
! debug
!
! ! Get local min and max of T5 coeff to avoid to parse over large tables
! !
! tmin_l = 0
! tmax_l = fmax_l + floor(real(lmax_l)/2)
GMAXX = min(pmax_l,2*mmax_l + 2*jmax_l)
HMAXX = jmax_l + mmax_l
! idMAXT5 = max(GMAXX,HMAXX)+1
!
!
ELSE ! ... if NPROC = 1
!
! Set local indices to global
idx_s_l = 1
idx_e_l = NbOfCoeffs
!
ALLOCATE(idxCoeffs(1:NbOfCoeffs,4))
idxCoeffs = 0
!
! Get idx from NNT5_filename
OPEN(fid,FILE='indices.in')
!
ENDOF = 0
EXT =0
NN = 1
!
DO WHILE (ENDOF .eq. 0 .and. EXT .eq. 0)
READ(fid,*,IOSTAT=ENDOF) IT1,IT2,IT3,IT4
IF( NN .ge. idx_s_l .and. NN .le. idx_e_l) THEN
! allocate local coefficient
idxCoeffs(NN,1) = IT1
idxCoeffs(NN,2) = IT2
idxCoeffs(NN,3) = IT3
idxCoeffs(NN,4) = IT4
ENDIF
NN = NN +1
ENDDO
!
CLOSE(fid)
! Set chunksize to global
nb_l = NbOfCoeffs
! !
! Get local maximum and minimum indices
pmin_l = minval(idxCoeffs(:,1))
pmax_l = maxval(idxCoeffs(:,1))
!
jmin_l = minval(idxCoeffs(:,2))
jmax_l = maxval(idxCoeffs(:,2))
!
mmin_l = minval(idxCoeffs(:,3))
mmax_l = maxval(idxCoeffs(:,3))
!
nmin_l = minval(idxCoeffs(:,4))
nmax_l = maxval(idxCoeffs(:,4))
!
! Get local min and max of T5 coeff to avoid to parse over large tables
!
! idMAXT5 = pmax_l + 2*jmax_l
! tmin_l = 0
! tmax_l = fmax_l + floor(real(lmax_l)/2)
ENDIF
!
END SUBROUTINE PPSETUP
c4science · Help