Skip to content

Add MPI routine to tell workers when a new iteration starts #45

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
MiCurry wants to merge 1 commit into
base: main
Choose a base branch
from
Open

Add MPI routine to tell workers when a new iteration starts #45

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
42 changes: 41 additions & 1 deletion f90/MPI/Main_MPI.f90
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -1455,6 +1455,40 @@ Subroutine Master_job_Distribute_userdef_control(ctrl, comm)

end Subroutine Master_job_Distribute_userdef_control


subroutine Master_job_send_inv_iteration(iteration_num, comm)

implicit none

integer, intent(in) :: iteration_num
integer, intent(in), optional :: comm
integer :: task, size_current, comm_current

if (present(comm)) then
if (comm .eq. MPI_COMM_NULL) then
comm_current = comm_world
else
comm_current = comm
endif
else
comm_current = comm_world
end if
call MPI_COMM_SIZE( comm_current, size_current, ierr )

worker_job_task % what_to_do = 'ITER_NUM'
worker_job_task % per_index = -1
worker_job_task % pol_index = -1
call create_worker_job_task_place_holder
call Pack_worker_job_task

do task = 1, size_current - 1
call MPI_SEND(worker_job_package, Nbytes, MPI_PACKED, task, FROM_MASTER, comm_current, ierr)
end do

write(0,*) "New Iteration: ", iteration_num

end subroutine Master_job_send_inv_iteration

!######################### Master_job_Clean Memory ##########################
Subroutine Master_job_Clean_Memory(comm)

Expand Down Expand Up @@ -1827,7 +1861,7 @@ Subroutine Worker_job(sigma,d)

Integer :: iper,ipol,i,des_index
Integer :: per_index,per_index_pre
Integer :: pol_index, stn_index
Integer :: pol_index, stn_index, iteration_number
Integer :: eAll_vec_size
Integer :: comm_current, rank_current
Integer :: cpu_only_ranks
Expand All @@ -1854,6 +1888,7 @@ Subroutine Worker_job(sigma,d)

nTx=d%nTx
recv_loop=0
iteration_number = 0
previous_message=''
write(node_info,'(a5,i3.3,a4)') 'node[',taskid,']: '

Expand Down Expand Up @@ -2565,6 +2600,11 @@ Subroutine Worker_job(sigma,d)
if (associated(group_sizes)) then
deallocate(group_sizes)
endif
elseif (trim(worker_job_task%what_to_do) .eq. 'ITER_NUM' ) then

iteration_number = iteration_number + 1
write(0,*) "New Iteration: ", iteration_number

elseif (trim(worker_job_task%what_to_do) .eq. 'STOP' ) then
! clear all the temp packages and stop
if (associated(sigma_para_vec)) then
Expand Down
Loading