10
0
mirror of https://github.com/LCPQ/quantum_package synced 2024-12-22 20:35:19 +01:00

openMP in MRCC_new and corected bug

This commit is contained in:
Anthony Scemama 2015-07-29 15:16:42 +02:00
parent ed7d11bf04
commit c042e5af0c

View File

@ -18,58 +18,103 @@ subroutine mrcc_dress(ndetref,ndetnonref,nstates,delta_ij_,delta_ii_)
double precision :: hij,phase
integer :: exc(0:2,2,2),degree
logical :: is_in_wavefunction
double precision, allocatable :: delta_ij_tmp(:,:,:), delta_ii_tmp(:,:)
i_state = 1
allocate(excitation_operators(5,N_det_non_ref))
allocate(amplitudes_phase_less(N_det_non_ref))
allocate(key_test(N_int,2))
allocate(index_connected(N_det_non_ref))
!$OMP PARALLEL DEFAULT(NONE) &
!$OMP SHARED(N_det_ref, N_det_non_ref, psi_ref, i_state, &
!$OMP N_connect_ref,index_connected,psi_non_ref, &
!$OMP excitation_operators,amplitudes_phase_less, &
!$OMP psi_non_ref_coef,N_int,lambda_mrcc,N_det, &
!$OMP delta_ii_,delta_ij_,psi_ref_coef,nstates) &
!$OMP PRIVATE(i,j,k,l,hil,phase_il,exc,degree,t_il, &
!$OMP key_test,i_ok,phase_la,hij,phase_ij, &
!$OMP dij,degree_vector,idx_vector,delta_ij_tmp, &
!$OMP delta_ii_tmp,phase)
allocate(idx_vector(0:N_det_non_ref))
allocate(degree_vector(N_det_non_ref))
i_state = 1
allocate(key_test(N_int,2))
allocate(delta_ij_tmp(size(delta_ij_,1),size(delta_ij_,2),nstates))
allocate(delta_ii_tmp(size(delta_ij_,1),nstates))
delta_ij_tmp = 0.d0
delta_ii_tmp = 0.d0
do i = 1, N_det_ref
!$OMP SINGLE
call get_excitation_operators_for_one_ref(psi_ref(1,1,i),i_state,N_det_non_ref,N_connect_ref,excitation_operators,amplitudes_phase_less,index_connected)
print*,'N_connect_ref =',N_connect_ref
print*,'N_det_non_ref =',N_det_non_ref
!$OMP END SINGLE
!$OMP BARRIER
!$OMP DO SCHEDULE(guided)
do l = 1, N_det_non_ref
double precision :: t_il,phase_il,hil
call i_H_j_phase_out(psi_ref(1,1,i),psi_non_ref(1,1,l),N_int,hil,phase_il,exc,degree)
t_il = hil * lambda_mrcc(i_state,l)
! loop on the non ref determinants
do j = 1, N_connect_ref
! loop on the excitation operators linked to i
if(j==l)cycle
do k = 1, N_int
key_test(k,1) = psi_non_ref(k,1,l)
key_test(k,2) = psi_non_ref(k,2,l)
enddo
! we apply the excitation operator T_I->j
call apply_excitation_operator(key_test,excitation_operators(1,j),i_ok)
if(i_ok.ne.1)cycle
! we check if such determinant is already in the wave function
if(is_in_wavefunction(key_test,N_int,N_det))cycle
! we get the phase for psi_non_ref(l) -> T_I->j |psi_non_ref(l)>
call get_excitation(psi_non_ref(1,1,l),key_test,exc,degree,phase_la,N_int)
! we get the phase T_I->j
call i_H_j_phase_out(psi_ref(1,1,i),psi_non_ref(1,1,index_connected(j)),N_int,hij,phase_ij,exc,degree)
! we compute the contribution to the coef of key_test
dij = t_il * hij * phase_la *phase_ij *lambda_mrcc(i_state,index_connected(j)) * 0.5d0
! we compute the interaction of such determinant with all the non_ref dets
call get_excitation_degree_vector(psi_non_ref,key_test,degree_vector,N_int,N_det_non_ref,idx_vector)
do k = 1, idx_vector(0)
call i_H_j_phase_out(key_test,psi_non_ref(1,1,idx_vector(k)),N_int,hij,phase,exc,degree)
delta_ij_(i,idx_vector(k),i_state) += hij * dij
delta_ij_tmp(i,idx_vector(k),i_state) += hij * dij
enddo
enddo
if(dabs(psi_ref_coef(i,i_state)).le.5.d-5)cycle
! delta_ij_(i,l,i_state) = delta_ij_(i,l,i_state) * 0.5d0
delta_ii_(i,i_state) -= delta_ij_(i,l,i_state) * psi_non_ref_coef(l,i_state) / psi_ref_coef(i,i_state)
if(dabs(psi_ref_coef(i,i_state)).le.5.d-5) then
delta_ii_tmp(i,i_state) -= &
delta_ij_tmp(i,l,i_state) * psi_non_ref_coef(l,i_state) &
/ psi_ref_coef(i,i_state)
endif
enddo
!$OMP END DO
enddo
!$OMP CRITICAL
delta_ij_ = delta_ij_ + delta_ij_tmp
delta_ii_ = delta_ii_ + delta_ii_tmp
!$OMP END CRITICAL
deallocate(delta_ii_tmp,delta_ij_tmp)
deallocate(idx_vector)
deallocate(key_test)
deallocate(degree_vector)
!$OMP END PARALLEL
deallocate(excitation_operators)
deallocate(amplitudes_phase_less)
deallocate(key_test)
deallocate(idx_vector)
deallocate(degree_vector)
end