9
1
mirror of https://github.com/QuantumPackage/qp2.git synced 2024-11-07 14:03:37 +01:00

Update mini in slaves

This commit is contained in:
Anthony Scemama 2019-01-31 17:28:54 +01:00
parent 8bd05b2c3f
commit d76f39527d
2 changed files with 8 additions and 5 deletions

View File

@ -103,14 +103,14 @@ subroutine run_pt2_slave(thread,iproc,energy)
done = .true. done = .true.
endif endif
call sort_selection_buffer(b) call sort_selection_buffer(b)
call push_pt2_results_async_recv(zmq_socket_push,sending) call push_pt2_results_async_recv(zmq_socket_push,b%mini,sending)
call push_pt2_results_async_send(zmq_socket_push, i_generator, pt2, variance, norm, b, task_id, n_tasks,sending) call push_pt2_results_async_send(zmq_socket_push, i_generator, pt2, variance, norm, b, task_id, n_tasks,sending)
b%cur=0 b%cur=0
! Try to adjust n_tasks around nproc/8 seconds per job ! Try to adjust n_tasks around nproc/8 seconds per job
n_tasks = min(2*n_tasks,int( dble(n_tasks * nproc/8) / (time1 - time0 + 1.d0))) n_tasks = min(2*n_tasks,int( dble(n_tasks * nproc/8) / (time1 - time0 + 1.d0)))
end do end do
call push_pt2_results_async_recv(zmq_socket_push,sending) call push_pt2_results_async_recv(zmq_socket_push,b%mini,sending)
integer, external :: disconnect_from_taskserver integer, external :: disconnect_from_taskserver
do i=1,300 do i=1,300
@ -321,7 +321,7 @@ subroutine push_pt2_results_async_send(zmq_socket_push, index, pt2, variance, no
end subroutine end subroutine
subroutine push_pt2_results_async_recv(zmq_socket_push,sending) subroutine push_pt2_results_async_recv(zmq_socket_push,mini,sending)
use f77_zmq use f77_zmq
use selection_types use selection_types
implicit none implicit none
@ -329,6 +329,7 @@ subroutine push_pt2_results_async_recv(zmq_socket_push,sending)
integer(ZMQ_PTR), intent(in) :: zmq_socket_push integer(ZMQ_PTR), intent(in) :: zmq_socket_push
integer(ZMQ_PTR), intent(inout) :: sending integer(ZMQ_PTR), intent(inout) :: sending
integer :: rc integer :: rc
double precision :: mini
if (.not.sending) return if (.not.sending) return
@ -343,6 +344,7 @@ IRP_ELSE
print *, irp_here//': error in receiving ok' print *, irp_here//': error in receiving ok'
stop -1 stop -1
endif endif
rc = f77_zmq_recv( zmq_socket_push, mini, 8, 0)
IRP_ENDIF IRP_ENDIF
sending = .False. sending = .False.
end subroutine end subroutine
@ -438,7 +440,7 @@ subroutine pull_pt2_results(zmq_socket_pull, index, pt2, variance, norm, task_id
! Activate is zmq_socket_pull is a REP ! Activate is zmq_socket_pull is a REP
IRP_IF ZMQ_PUSH IRP_IF ZMQ_PUSH
IRP_ELSE IRP_ELSE
rc = f77_zmq_send( zmq_socket_pull, 'ok', 2, 0) rc = f77_zmq_send( zmq_socket_pull, 'ok', 2, ZMQ_SNDMORE)
if (rc == -1) then if (rc == -1) then
n_tasks = 1 n_tasks = 1
task_id(1) = 0 task_id(1) = 0
@ -446,6 +448,7 @@ IRP_ELSE
print *, irp_here//': error in sending ok' print *, irp_here//': error in sending ok'
stop -1 stop -1
endif endif
rc = f77_zmq_send( zmq_socket_pull, b%mini, 8, 0)
IRP_ENDIF IRP_ENDIF
end subroutine end subroutine

View File

@ -10,7 +10,7 @@ subroutine run_stochastic_cipsi
double precision :: rss double precision :: rss
double precision, external :: memory_of_double double precision, external :: memory_of_double
PROVIDE H_apply_buffer_allocated PROVIDE H_apply_buffer_allocated N_generators_bitmask
threshold_generators = 1.d0 threshold_generators = 1.d0
SOFT_TOUCH threshold_generators SOFT_TOUCH threshold_generators