10
0
mirror of https://github.com/LCPQ/quantum_package synced 2024-11-19 04:22:36 +01:00
quantum_package/plugins/Full_CI_ZMQ/broadcast.irp.f

73 lines
1.9 KiB
Fortran
Raw Normal View History

2017-11-23 11:16:38 +01:00
subroutine mpi_bcast_psi(energy, size_energy)
implicit none
BEGIN_DOC
! Broadcast the wave function via MPI
END_DOC
integer, intent(in) :: size_energy
double precision, intent(inout) :: energy(size_energy)
PROVIDE mpi_initialized
IRP_IF MPI
include 'mpif.h'
integer :: ierr
call MPI_BCAST (N_states, 1, MPI_INTEGER, 0, MPI_COMM_WORLD, ierr)
2017-11-23 18:07:24 +01:00
if (ierr /= MPI_SUCCESS) then
2017-11-23 21:12:41 +01:00
print *, irp_here//': Unable to broadcast N_states'
2017-11-23 11:16:38 +01:00
stop -1
endif
call MPI_BCAST (N_det, 1, MPI_INTEGER, 0, MPI_COMM_WORLD, ierr)
2017-11-23 18:07:24 +01:00
if (ierr /= MPI_SUCCESS) then
2017-11-23 21:12:41 +01:00
print *, irp_here//': Unable to broadcast N_det'
2017-11-23 11:16:38 +01:00
stop -1
endif
call MPI_BCAST (psi_det_size, 1, MPI_INTEGER, 0, MPI_COMM_WORLD, ierr)
2017-11-23 18:07:24 +01:00
if (ierr /= MPI_SUCCESS) then
2017-11-23 21:12:41 +01:00
print *, irp_here//': Unable to broadcast psi_det_size'
2017-11-23 11:16:38 +01:00
stop -1
endif
if (.not.mpi_master) then
TOUCH psi_det_size N_det N_states
endif
2017-11-23 18:07:24 +01:00
2017-11-23 19:24:42 +01:00
call broadcast_chunks_bit_kind(psi_det,N_det*N_int*2)
call broadcast_chunks_double(psi_coef,N_states*N_det)
2017-11-23 11:16:38 +01:00
if (.not.mpi_master) then
TOUCH psi_det psi_coef
endif
call MPI_BCAST (N_det_generators, 1, MPI_INTEGER, 0, MPI_COMM_WORLD, ierr)
2017-11-23 18:07:24 +01:00
if (ierr /= MPI_SUCCESS) then
2017-11-23 21:12:41 +01:00
print *, irp_here//': Unable to broadcast N_det_generators'
2017-11-23 11:16:38 +01:00
stop -1
endif
if (.not.mpi_master) then
TOUCH N_det_generators
endif
call MPI_BCAST (N_det_selectors, 1, MPI_INTEGER, 0, MPI_COMM_WORLD, ierr)
2017-11-23 18:07:24 +01:00
if (ierr /= MPI_SUCCESS) then
2017-11-23 21:12:41 +01:00
print *, irp_here//': Unable to broadcast N_det_selectors'
2017-11-23 11:16:38 +01:00
stop -1
endif
if (.not.mpi_master) then
TOUCH N_det_selectors
endif
call MPI_BCAST (energy, size(energy), MPI_DOUBLE_PRECISION, 0, MPI_COMM_WORLD, ierr)
2017-11-23 18:07:24 +01:00
if (ierr /= MPI_SUCCESS) then
2017-11-23 21:12:41 +01:00
print *, irp_here//': Unable to broadcast energy'
2017-11-23 11:16:38 +01:00
stop -1
endif
IRP_ENDIF
end