99 do_hfx, do_admm, qs_env, kernel_env, kernel_env_admm_aux, &
100 sub_env, work_matrices, admm_symm, admm_xc_correction, do_lrigpw, &
102 TYPE(
cp_fm_type),
DIMENSION(:, :),
INTENT(INOUT) :: aop_evects
103 TYPE(
cp_fm_type),
DIMENSION(:, :),
INTENT(IN) :: evects
104 LOGICAL,
INTENT(in) :: is_rks_triplets, do_hfx, do_admm
109 LOGICAL,
INTENT(in) :: admm_symm, admm_xc_correction, &
110 do_lrigpw, tddfpt_mgrid
112 CHARACTER(LEN=*),
PARAMETER :: routinen =
'fhxc_kernel'
114 CHARACTER(LEN=default_string_length) :: basis_type
115 INTEGER :: handle, ikind, ispin, ivect, nao, &
116 nao_aux, nkind, nspins, nvects, &
118 INTEGER,
DIMENSION(:),
POINTER :: blk_sizes
119 INTEGER,
DIMENSION(maxspins) :: nactive
120 LOGICAL :: do_noncol, do_onecenter, do_scale, gapw, &
123 TYPE(
cp_fm_type) :: work_aux_orb, work_orb_orb
124 TYPE(
dbcsr_p_type),
DIMENSION(:),
POINTER :: a_xc_munu_sub, rho_ia_ao, &
130 TYPE(
pw_c1d_gs_type),
DIMENSION(:),
POINTER :: rho_ia_g, rho_ia_g_aux_fit
133 TYPE(
pw_r3d_rs_type),
DIMENSION(:),
POINTER :: rho_ia_r, rho_ia_r_aux_fit
135 TYPE(
rho_atom_type),
DIMENSION(:),
POINTER :: rho1_atom_set, rho_atom_set
138 CALL timeset(routinen, handle)
140 nspins =
SIZE(evects, 1)
141 nvects =
SIZE(evects, 2)
144 cpassert(
ASSOCIATED(sub_env%admm_A))
146 CALL get_qs_env(qs_env, dft_control=dft_control)
148 gapw = dft_control%qs_control%gapw
149 gapw_xc = dft_control%qs_control%gapw_xc
150 spinflip = dft_control%tddfpt2_control%spinflip
153 do_scale = is_rks_triplets
154 do_onecenter = gapw .OR. gapw_xc
161 CALL qs_rho_get(work_matrices%rho_orb_struct_sub, rho_ao=rho_ia_ao, &
162 rho_g=rho_ia_g, rho_r=rho_ia_r)
163 IF (do_hfx .AND. do_admm)
THEN
165 CALL qs_rho_get(work_matrices%rho_aux_fit_struct_sub, &
166 rho_ao=rho_ia_ao_aux_fit, rho_g=rho_ia_g_aux_fit, &
167 rho_r=rho_ia_r_aux_fit)
171 CALL get_qs_env(qs_env, xcint_weights=weights)
176 IF (
ALLOCATED(work_matrices%evects_sub))
THEN
177 IF (
ASSOCIATED(work_matrices%evects_sub(1, ivect)%matrix_struct))
THEN
179 CALL dbcsr_set(rho_ia_ao(ispin)%matrix, 0.0_dp)
181 matrix_v=sub_env%mos_active(ispin), &
182 matrix_g=work_matrices%evects_sub(ispin, ivect), &
183 ncol=nactive(ispin), symmetry_mode=1)
191 CALL dbcsr_set(rho_ia_ao(ispin)%matrix, 0.0_dp)
193 matrix_v=sub_env%mos_active(ispin), &
194 matrix_g=evects(ispin, ivect), &
195 ncol=nactive(ispin), symmetry_mode=1)
201 pw_env_external=sub_env%pw_env, &
202 task_list_external=sub_env%task_list_orb, &
203 para_env_external=sub_env%para_env, &
204 tddfpt_lri_env=kernel_env%lri_env, &
205 tddfpt_lri_density=kernel_env%lri_density)
206 ELSE IF (dft_control%qs_control%lrigpw .OR. &
207 dft_control%qs_control%rigpw)
THEN
209 pw_env_external=sub_env%pw_env, &
210 task_list_external=sub_env%task_list_orb, &
211 para_env_external=sub_env%para_env)
215 local_rho_set=work_matrices%local_rho_set, &
216 pw_env_external=sub_env%pw_env, &
217 task_list_external=sub_env%task_list_orb_soft, &
218 para_env_external=sub_env%para_env)
220 do_rho0=(.NOT. is_rks_triplets), pw_env_sub=sub_env%pw_env)
221 ELSE IF (gapw_xc)
THEN
223 rho_xc_external=work_matrices%rho_xc_struct_sub, &
224 local_rho_set=work_matrices%local_rho_set, &
225 pw_env_external=sub_env%pw_env, &
226 task_list_external=sub_env%task_list_orb, &
227 task_list_external_soft=sub_env%task_list_orb_soft, &
228 para_env_external=sub_env%para_env)
229 CALL prepare_gapw_den(qs_env, work_matrices%local_rho_set, do_rho0=.false., &
230 pw_env_sub=sub_env%pw_env)
233 pw_env_external=sub_env%pw_env, &
234 task_list_external=sub_env%task_list_orb, &
235 para_env_external=sub_env%para_env)
240 CALL dbcsr_set(work_matrices%A_ia_munu_sub(ispin)%matrix, 0.0_dp)
245 CALL pw_zero(work_matrices%A_ia_rspace_sub(ispin))
250 IF ((.NOT. dft_control%tddfpt2_control%do_bse) .AND. (.NOT. dft_control%tddfpt2_control%do_bse_w_only))
THEN
251 IF ((.NOT. dft_control%tddfpt2_control%do_bse_gw_only))
THEN
255 CALL pw_zero(work_matrices%wpw_rspace_sub(ispin))
256 IF (
ASSOCIATED(work_matrices%wpw_tau_rspace_sub))
CALL pw_zero(work_matrices%wpw_tau_rspace_sub(ispin))
258 IF (do_onecenter)
THEN
259 rho_atom_set => sub_env%local_rho_set%rho_atom_set
260 rho1_atom_set => work_matrices%local_rho_set%rho_atom_set
264 IF (kernel_env%do_exck)
THEN
267 CALL qs_fxc_apply(qs_env, kernel_env%xc_deriv_set, kernel_env%xc_rho_set, &
268 work_matrices%rho_xc_struct_sub, rho_atom_set, &
269 kernel_env%xc_section, do_onecenter, work_matrices%wpw_rspace_sub, &
270 work_matrices%wpw_tau_rspace_sub, rho1_atom_set, &
271 do_scale=do_scale, is_triplet=is_rks_triplets, spinflip=do_noncol, &
272 pw_env_ext=sub_env%pw_env, para_env_external=sub_env%para_env)
274 CALL pw_axpy(work_matrices%wpw_rspace_sub(ispin), &
275 work_matrices%A_ia_rspace_sub(ispin), kernel_env%alpha)
280 CALL pw_scale(work_matrices%A_ia_rspace_sub(ispin), &
281 work_matrices%A_ia_rspace_sub(ispin)%pw_grid%dvol)
282 CALL integrate_v_rspace(v_rspace=work_matrices%A_ia_rspace_sub(ispin), &
283 hmat=work_matrices%A_ia_munu_sub(ispin), &
284 qs_env=qs_env, calculate_forces=.false., gapw=gapw_xc, &
285 pw_env_external=sub_env%pw_env, &
286 task_list_external=sub_env%task_list_orb_soft)
287 CALL pw_zero(work_matrices%A_ia_rspace_sub(ispin))
290 IF (kernel_env%do_exck)
THEN
292 work_matrices%rho_orb_struct_sub, is_rks_triplets)
294 CALL qs_fxc_apply(qs_env, kernel_env%xc_deriv_set, kernel_env%xc_rho_set, &
295 work_matrices%rho_orb_struct_sub, rho_atom_set, &
296 kernel_env%xc_section, do_onecenter, work_matrices%wpw_rspace_sub, &
297 work_matrices%wpw_tau_rspace_sub, rho1_atom_set, &
298 do_scale=do_scale, is_triplet=is_rks_triplets, spinflip=do_noncol, &
299 pw_env_ext=sub_env%pw_env, para_env_external=sub_env%para_env)
301 CALL pw_axpy(work_matrices%wpw_rspace_sub(ispin), &
302 work_matrices%A_ia_rspace_sub(ispin), kernel_env%alpha)
313 IF ((.NOT. dft_control%tddfpt2_control%do_bse) .AND. (.NOT. dft_control%tddfpt2_control%do_bse_w_only) &
314 .AND. (.NOT. dft_control%tddfpt2_control%do_bse_gw_only))
THEN
315 IF (do_admm .AND. admm_xc_correction)
THEN
318 rho_aux_fit_struct=work_matrices%rho_aux_fit_struct_sub, &
319 local_rho_set=work_matrices%local_rho_set_admm, &
320 qs_env=qs_env, sub_env=sub_env, &
321 wfm_rho_orb=work_matrices%rho_ao_orb_fm_sub, &
322 wfm_rho_aux_fit=work_matrices%rho_ao_aux_fit_fm_sub, &
323 wfm_aux_orb=work_matrices%wfm_aux_orb_sub)
326 CALL dbcsr_get_info(rho_ia_ao_aux_fit(1)%matrix, row_blk_size=blk_sizes)
327 ALLOCATE (a_xc_munu_sub(nspins))
329 ALLOCATE (a_xc_munu_sub(ispin)%matrix)
330 CALL dbcsr_create(matrix=a_xc_munu_sub(ispin)%matrix, name=
"ADMM_XC", &
331 dist=sub_env%dbcsr_dist, matrix_type=dbcsr_type_symmetric, &
332 row_blk_size=blk_sizes, col_blk_size=blk_sizes)
334 CALL dbcsr_set(a_xc_munu_sub(ispin)%matrix, 0.0_dp)
337 CALL pw_env_get(sub_env%pw_env, auxbas_pw_pool=auxbas_pw_pool)
338 ALLOCATE (v_rspace_sub(nspins))
340 CALL auxbas_pw_pool%create_pw(v_rspace_sub(ispin))
341 CALL pw_zero(v_rspace_sub(ispin))
344 IF (admm_env%do_gapw)
THEN
345 basis_type =
"AUX_FIT_SOFT"
346 task_list => sub_env%task_list_aux_fit_soft
348 basis_type =
"AUX_FIT"
349 task_list => sub_env%task_list_aux_fit
352 IF (admm_env%do_gapw)
THEN
353 rho_atom_set => sub_env%local_rho_set_admm%rho_atom_set
354 rho1_atom_set => work_matrices%local_rho_set_admm%rho_atom_set
357 CALL pw_zero(work_matrices%wpw_rspace_sub(ispin))
358 IF (
ASSOCIATED(work_matrices%wpw_tau_rspace_sub))
THEN
359 CALL pw_zero(work_matrices%wpw_tau_rspace_sub(ispin))
363 CALL qs_fxc_apply(qs_env, kernel_env_admm_aux%xc_deriv_set, kernel_env_admm_aux%xc_rho_set, &
364 work_matrices%rho_aux_fit_struct_sub, rho_atom_set, &
365 kernel_env_admm_aux%xc_section, admm_env%do_gapw, &
366 work_matrices%wpw_rspace_sub, work_matrices%wpw_tau_rspace_sub, rho1_atom_set, &
367 do_scale=do_scale, is_triplet=is_rks_triplets, spinflip=do_noncol, &
368 pw_env_ext=sub_env%pw_env, kind_set_external=admm_env%admm_gapw_env%admm_kind_set, &
369 para_env_external=sub_env%para_env)
371 CALL pw_axpy(work_matrices%wpw_rspace_sub(ispin), v_rspace_sub(ispin), &
372 kernel_env_admm_aux%alpha)
376 CALL pw_scale(v_rspace_sub(ispin), v_rspace_sub(ispin)%pw_grid%dvol)
377 CALL integrate_v_rspace(v_rspace=v_rspace_sub(ispin), &
378 hmat=a_xc_munu_sub(ispin), &
379 qs_env=qs_env, calculate_forces=.false., &
380 pw_env_external=sub_env%pw_env, &
381 basis_type=basis_type, &
382 task_list_external=task_list)
384 IF (admm_env%do_gapw)
THEN
385 CALL update_ks_atom(qs_env, a_xc_munu_sub, rho_ia_ao_aux_fit, forces=.false., tddft=.true., &
386 rho_atom_external=rho1_atom_set, &
387 kind_set_external=admm_env%admm_gapw_env%admm_kind_set, &
388 oce_external=admm_env%admm_gapw_env%oce, &
389 sab_external=sub_env%sab_aux_fit)
392 CALL dbcsr_create(dbwork, template=work_matrices%A_ia_munu_sub(1)%matrix)
394 matrix_struct=work_matrices%wfm_aux_orb_sub%matrix_struct)
396 matrix_struct=work_matrices%rho_ao_orb_fm_sub%matrix_struct)
397 CALL cp_fm_get_info(work_aux_orb, nrow_global=nao_aux, ncol_global=nao)
401 CALL parallel_gemm(
'T',
'N', nao, nao, nao_aux, 1.0_dp, sub_env%admm_A, &
402 work_aux_orb, 0.0_dp, work_orb_orb)
403 CALL dbcsr_copy(dbwork, work_matrices%A_ia_munu_sub(1)%matrix)
406 CALL dbcsr_add(work_matrices%A_ia_munu_sub(ispin)%matrix, dbwork, 1.0_dp, 1.0_dp)
411 CALL auxbas_pw_pool%give_back_pw(v_rspace_sub(ispin))
413 DEALLOCATE (v_rspace_sub)
419 DEALLOCATE (a_xc_munu_sub)
421 IF (admm_env%do_gapw)
THEN
422 cpwarn(
"GAPW/ADMM needs symmetric ADMM kernel")
423 cpabort(
"GAPW/ADMM@TDDFT")
426 CALL pw_zero(work_matrices%wpw_rspace_sub(ispin))
427 IF (
ASSOCIATED(work_matrices%wpw_tau_rspace_sub))
THEN
428 CALL pw_zero(work_matrices%wpw_tau_rspace_sub(ispin))
432 CALL qs_fxc_apply(qs_env, kernel_env_admm_aux%xc_deriv_set, kernel_env_admm_aux%xc_rho_set, &
433 work_matrices%rho_aux_fit_struct_sub, rho_atom_set, &
434 kernel_env_admm_aux%xc_section, admm_env%do_gapw, &
435 work_matrices%wpw_rspace_sub, work_matrices%wpw_tau_rspace_sub, rho1_atom_set, &
436 do_scale=do_scale, is_triplet=is_rks_triplets, spinflip=do_noncol, &
437 pw_env_ext=sub_env%pw_env)
440 CALL pw_axpy(work_matrices%wpw_rspace_sub(ispin), &
441 work_matrices%A_ia_rspace_sub(ispin), kernel_env_admm_aux%alpha)
450 IF (.NOT. dft_control%tddfpt2_control%do_bse_w_only)
THEN
451 IF (.NOT. dft_control%tddfpt2_control%do_bse_gw_only)
THEN
452 IF ((.NOT. is_rks_triplets) .AND. (spinflip ==
no_sf_tddfpt))
THEN
457 CALL pw_axpy(rho_ia_g(ispin), rho_ia_g(1))
460 rho_ia_g=rho_ia_g(1), &
461 local_rho_set=work_matrices%local_rho_set, &
462 hartree_local=work_matrices%hartree_local, &
463 qs_env=qs_env, sub_env=sub_env, gapw=gapw, &
464 work_v_gspace=work_matrices%wpw_gspace_sub(1), &
465 work_v_rspace=work_matrices%wpw_rspace_sub(1), &
466 tddfpt_mgrid=tddfpt_mgrid)
471 IF (.NOT. do_lrigpw)
THEN
472 CALL pw_scale(work_matrices%A_ia_rspace_sub(ispin), &
473 work_matrices%A_ia_rspace_sub(ispin)%pw_grid%dvol)
476 CALL integrate_v_rspace(v_rspace=work_matrices%A_ia_rspace_sub(ispin), &
477 hmat=work_matrices%A_ia_munu_sub(ispin), &
478 qs_env=qs_env, calculate_forces=.false., gapw=gapw, &
479 pw_env_external=sub_env%pw_env, &
480 task_list_external=sub_env%task_list_orb_soft)
481 ELSE IF (gapw_xc)
THEN
482 IF (.NOT. is_rks_triplets)
THEN
483 CALL integrate_v_rspace(v_rspace=work_matrices%A_ia_rspace_sub(ispin), &
484 hmat=work_matrices%A_ia_munu_sub(ispin), &
485 qs_env=qs_env, calculate_forces=.false., gapw=.false., &
486 pw_env_external=sub_env%pw_env, task_list_external=sub_env%task_list_orb)
489 CALL integrate_v_rspace(v_rspace=work_matrices%A_ia_rspace_sub(ispin), &
490 hmat=work_matrices%A_ia_munu_sub(ispin), &
491 qs_env=qs_env, calculate_forces=.false., gapw=.false., &
492 pw_env_external=sub_env%pw_env, task_list_external=sub_env%task_list_orb)
495 CALL pw_scale(work_matrices%A_ia_rspace_sub(ispin), &
496 work_matrices%A_ia_rspace_sub(ispin)%pw_grid%dvol)
497 lri_v_int => kernel_env%lri_density%lri_coefs(ispin)%lri_kinds
498 CALL get_qs_env(qs_env, nkind=nkind, para_env=para_env)
500 lri_v_int(ikind)%v_int = 0.0_dp
502 CALL integrate_v_rspace_one_center(work_matrices%A_ia_rspace_sub(ispin), &
503 qs_env, lri_v_int, .false.,
"P_LRI_AUX")
505 CALL para_env%sum(lri_v_int(ikind)%v_int)
511 IF (.NOT. do_lrigpw)
THEN
512 IF (gapw .OR. gapw_xc)
THEN
514 CALL update_ks_atom(qs_env, work_matrices%A_ia_munu_sub, rho_ia_ao, forces=.false., &
515 rho_atom_external=work_matrices%local_rho_set%rho_atom_set, &
522 IF (do_lrigpw .AND. (.NOT. is_rks_triplets))
THEN
523 CALL tddfpt2_lri_amat(qs_env, sub_env, kernel_env%lri_env, lri_v_int, work_matrices%A_ia_munu_sub)
527 IF (
ALLOCATED(work_matrices%evects_sub))
THEN
529 sub_env%mos_active(ispin), &
530 work_matrices%Aop_evects_sub(ispin, ivect), &
531 ncol=nactive(ispin), alpha=1.0_dp, beta=0.0_dp)
534 sub_env%mos_active(ispin), &
535 aop_evects(ispin, ivect), &
536 ncol=nactive(ispin), alpha=1.0_dp, beta=0.0_dp)
543 CALL timestop(handle)
subroutine, public get_qs_env(qs_env, atomic_kind_set, qs_kind_set, cell, super_cell, cell_ref, use_ref_cell, kpoints, dft_control, mos, sab_orb, sab_all, qmmm, qmmm_periodic, mimic, sac_ae, sac_ppl, sac_lri, sap_ppnl, sab_vdw, sab_scp, sap_oce, sab_lrc, sab_se, sab_xtbe, sab_tbe, sab_core, sab_xb, sab_xtb_pp, sab_xtb_nonbond, sab_almo, sab_kp, sab_kp_nosym, sab_cneo, particle_set, energy, force, matrix_h, matrix_h_im, matrix_ks, matrix_ks_im, matrix_vxc, run_rtp, rtp, matrix_h_kp, matrix_h_im_kp, matrix_ks_kp, matrix_ks_im_kp, matrix_vxc_kp, kinetic_kp, matrix_s_kp, matrix_w_kp, matrix_s_ri_aux_kp, matrix_s, matrix_s_ri_aux, matrix_w, matrix_p_mp2, matrix_p_mp2_admm, matrix_vhxc, rho, rho_xc, pw_env, ewald_env, ewald_pw, active_space, mpools, input, para_env, blacs_env, scf_control, rel_control, kinetic, qs_charges, vppl, xcint_weights, rho_core, rho_nlcc, rho_nlcc_g, ks_env, ks_qmmm_env, wf_history, scf_env, local_particles, local_molecules, distribution_2d, dbcsr_dist, molecule_kind_set, molecule_set, subsys, cp_subsys, oce, local_rho_set, rho_atom_set, task_list, task_list_soft, rho0_atom_set, rho0_mpole, rhoz_set, rhoz_cneo_set, ecoul_1c, rho0_s_rs, rho0_s_gs, rhoz_cneo_s_rs, rhoz_cneo_s_gs, do_kpoints, has_unit_metric, requires_mo_derivs, mo_derivs, mo_loc_history, nkind, natom, nelectron_total, nelectron_spin, efield, neighbor_list_id, linres_control, xas_env, virial, cp_ddapc_env, cp_ddapc_ewald, outer_scf_history, outer_scf_ihistory, x_data, et_coupling, dftb_potential, results, se_taper, se_store_int_env, se_nddo_mpole, se_nonbond_env, admm_env, lri_env, lri_density, exstate_env, ec_env, harris_env, dispersion_env, gcp_env, vee, rho_external, external_vxc, mask, mp2_env, bs_env, kg_env, wanniercentres, atprop, ls_scf_env, do_transport, transport_env, v_hartree_rspace, s_mstruct_changed, rho_changed, potential_changed, forces_up_to_date, mscfg_env, almo_scf_env, gradient_history, variable_history, embed_pot, spin_embed_pot, polar_env, mos_last_converged, eeq, rhs, do_rixs, tb_tblite)
Get the QUICKSTEP environment.