Skip to content

GitLab

fleur
fleur
Commit 9d4bdd07 authored by Uliana Alekseeva's avatar Uliana Alekseeva
Browse files
Options

Improved the load balance in init/stepf.F90

parent 558f834b
Hide whitespace changes
Inline Side-by-side
Showing with 10 additions and 5 deletions
init/stepf.F90
View file @ 9d4bdd07
...@@ -42,7 +42,7 @@ ...@@ -42,7 +42,7 @@
REAL g(3),gm(3),fJ REAL g(3),gm(3),fJ
REAL, ALLOCATABLE :: bfft(:) REAL, ALLOCATABLE :: bfft(:)
INTEGER, ALLOCATABLE :: icm(:,:,:) INTEGER, ALLOCATABLE :: icm(:,:,:)
INTEGER :: mpi_id, i3_start, i3_end INTEGER :: mpi_id, i3_start, i3_end, chunk_size,leftover_size
#ifdef CPP_MPI #ifdef CPP_MPI
INCLUDE 'mpif.h' INCLUDE 'mpif.h'
INTEGER ierr INTEGER ierr
...@@ -198,12 +198,17 @@ ...@@ -198,12 +198,17 @@
i3_end = im3 i3_end = im3
#ifdef CPP_MPI #ifdef CPP_MPI
IF (PRESENT(mpi)) THEN IF (PRESENT(mpi)) THEN
i3_start = mpi%irank * (im3/mpi%isize) chunk_size = im3/mpi%isize
i3_end = (mpi%irank + 1) * (im3/mpi%isize) - 1 leftover_size = modulo(im3,mpi%isize)
IF (mpi%irank == mpi%isize - 1) i3_end = im3 IF (mpi%irank < leftover_size ) THEN
i3_start = mpi%irank * (chunk_size + 1)
i3_end = (mpi%irank + 1) * (chunk_size + 1) - 1
ELSE
i3_start = leftover_size * (chunk_size + 1) + (mpi%irank - leftover_size) * chunk_size
i3_end = (i3_start + chunk_size) - 1
ENDIF
ENDIF ENDIF
#endif #endif
DO i3=i3_start,i3_end DO i3=i3_start,i3_end
gm(3)=REAL(i3) gm(3)=REAL(i3)
DO i2=0,3*stars%mx2-1 DO i2=0,3*stars%mx2-1
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment