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
132 TYPE(
pw_r3d_rs_type),
DIMENSION(:),
POINTER :: fxc_tau, rho_ia_r, rho_ia_r_aux_fit
134 TYPE(
rho_atom_type),
DIMENSION(:),
POINTER :: rho1_atom_set, rho_atom_set
137 CALL timeset(routinen, handle)
139 nspins =
SIZE(evects, 1)
140 nvects =
SIZE(evects, 2)
143 cpassert(
ASSOCIATED(sub_env%admm_A))
145 CALL get_qs_env(qs_env, dft_control=dft_control)
147 gapw = dft_control%qs_control%gapw
148 gapw_xc = dft_control%qs_control%gapw_xc
149 spinflip = dft_control%tddfpt2_control%spinflip
152 do_scale = is_rks_triplets
153 do_onecenter = gapw .OR. gapw_xc
160 CALL qs_rho_get(work_matrices%rho_orb_struct_sub, rho_ao=rho_ia_ao, &
161 rho_g=rho_ia_g, rho_r=rho_ia_r)
162 IF (do_hfx .AND. do_admm)
THEN
164 CALL qs_rho_get(work_matrices%rho_aux_fit_struct_sub, &
165 rho_ao=rho_ia_ao_aux_fit, rho_g=rho_ia_g_aux_fit, &
166 rho_r=rho_ia_r_aux_fit)
170 CALL get_qs_env(qs_env, xcint_weights=weights)
171 CALL pw_env_get(sub_env%pw_env, auxbas_pw_pool=auxbas_pw_pool)
177 IF (
ALLOCATED(work_matrices%evects_sub))
THEN
178 IF (
ASSOCIATED(work_matrices%evects_sub(1, ivect)%matrix_struct))
THEN
180 CALL dbcsr_set(rho_ia_ao(ispin)%matrix, 0.0_dp)
182 matrix_v=sub_env%mos_active(ispin), &
183 matrix_g=work_matrices%evects_sub(ispin, ivect), &
184 ncol=nactive(ispin), symmetry_mode=1)
192 CALL dbcsr_set(rho_ia_ao(ispin)%matrix, 0.0_dp)
194 matrix_v=sub_env%mos_active(ispin), &
195 matrix_g=evects(ispin, ivect), &
196 ncol=nactive(ispin), symmetry_mode=1)
202 pw_env_external=sub_env%pw_env, &
203 task_list_external=sub_env%task_list_orb, &
204 para_env_external=sub_env%para_env, &
205 tddfpt_lri_env=kernel_env%lri_env, &
206 tddfpt_lri_density=kernel_env%lri_density)
207 ELSE IF (dft_control%qs_control%lrigpw .OR. &
208 dft_control%qs_control%rigpw)
THEN
210 pw_env_external=sub_env%pw_env, &
211 task_list_external=sub_env%task_list_orb, &
212 para_env_external=sub_env%para_env)
216 local_rho_set=work_matrices%local_rho_set, &
217 pw_env_external=sub_env%pw_env, &
218 task_list_external=sub_env%task_list_orb_soft, &
219 para_env_external=sub_env%para_env)
221 do_rho0=(.NOT. is_rks_triplets), pw_env_sub=sub_env%pw_env)
222 ELSE IF (gapw_xc)
THEN
224 rho_xc_external=work_matrices%rho_xc_struct_sub, &
225 local_rho_set=work_matrices%local_rho_set, &
226 pw_env_external=sub_env%pw_env, &
227 task_list_external=sub_env%task_list_orb, &
228 task_list_external_soft=sub_env%task_list_orb_soft, &
229 para_env_external=sub_env%para_env)
230 CALL prepare_gapw_den(qs_env, work_matrices%local_rho_set, do_rho0=.false., &
231 pw_env_sub=sub_env%pw_env)
234 pw_env_external=sub_env%pw_env, &
235 task_list_external=sub_env%task_list_orb, &
236 para_env_external=sub_env%para_env)
241 CALL dbcsr_set(work_matrices%A_ia_munu_sub(ispin)%matrix, 0.0_dp)
246 CALL pw_zero(work_matrices%A_ia_rspace_sub(ispin))
251 IF ((.NOT. dft_control%tddfpt2_control%do_bse) .AND. (.NOT. dft_control%tddfpt2_control%do_bse_w_only))
THEN
252 IF ((.NOT. dft_control%tddfpt2_control%do_bse_gw_only))
THEN
255 IF (do_onecenter)
THEN
256 rho_atom_set => sub_env%local_rho_set%rho_atom_set
257 rho1_atom_set => work_matrices%local_rho_set%rho_atom_set
261 IF (kernel_env%do_exck)
THEN
264 CALL qs_fxc_apply(qs_env, kernel_env%xc_deriv_set, kernel_env%xc_rho_set, &
265 work_matrices%rho_xc_struct_sub, rho_atom_set, &
266 kernel_env%xc_section, do_onecenter, work_matrices%wpw_rspace_sub, &
267 fxc_tau, rho1_atom_set, &
268 do_scale=do_scale, is_triplet=is_rks_triplets, spinflip=do_noncol, &
269 pw_env_ext=sub_env%pw_env, para_env_external=sub_env%para_env)
271 CALL pw_axpy(work_matrices%wpw_rspace_sub(ispin), &
272 work_matrices%A_ia_rspace_sub(ispin), kernel_env%alpha)
277 CALL pw_scale(work_matrices%A_ia_rspace_sub(ispin), &
278 work_matrices%A_ia_rspace_sub(ispin)%pw_grid%dvol)
279 CALL integrate_v_rspace(v_rspace=work_matrices%A_ia_rspace_sub(ispin), &
280 hmat=work_matrices%A_ia_munu_sub(ispin), &
281 qs_env=qs_env, calculate_forces=.false., gapw=gapw_xc, &
282 pw_env_external=sub_env%pw_env, &
283 task_list_external=sub_env%task_list_orb_soft)
284 CALL pw_zero(work_matrices%A_ia_rspace_sub(ispin))
287 IF (kernel_env%do_exck)
THEN
289 work_matrices%rho_orb_struct_sub, is_rks_triplets)
291 CALL qs_fxc_apply(qs_env, kernel_env%xc_deriv_set, kernel_env%xc_rho_set, &
292 work_matrices%rho_orb_struct_sub, rho_atom_set, &
293 kernel_env%xc_section, do_onecenter, work_matrices%wpw_rspace_sub, &
294 fxc_tau, rho1_atom_set, &
295 do_scale=do_scale, is_triplet=is_rks_triplets, spinflip=do_noncol, &
296 pw_env_ext=sub_env%pw_env, para_env_external=sub_env%para_env)
298 CALL pw_axpy(work_matrices%wpw_rspace_sub(ispin), &
299 work_matrices%A_ia_rspace_sub(ispin), kernel_env%alpha)
311 IF (
ASSOCIATED(fxc_tau))
THEN
312 task_list => sub_env%task_list_orb
313 IF (do_onecenter) task_list => sub_env%task_list_orb_soft
315 CALL pw_scale(fxc_tau(ispin), kernel_env%alpha*fxc_tau(ispin)%pw_grid%dvol)
316 CALL integrate_v_rspace(v_rspace=fxc_tau(ispin), hmat=work_matrices%A_ia_munu_sub(ispin), &
317 qs_env=qs_env, calculate_forces=.false., compute_tau=.true., gapw=do_onecenter, &
318 pw_env_external=sub_env%pw_env, task_list_external=task_list)
319 CALL auxbas_pw_pool%give_back_pw(fxc_tau(ispin))
325 IF ((.NOT. dft_control%tddfpt2_control%do_bse) .AND. (.NOT. dft_control%tddfpt2_control%do_bse_w_only) &
326 .AND. (.NOT. dft_control%tddfpt2_control%do_bse_gw_only))
THEN
327 IF (do_admm .AND. admm_xc_correction)
THEN
330 rho_aux_fit_struct=work_matrices%rho_aux_fit_struct_sub, &
331 local_rho_set=work_matrices%local_rho_set_admm, &
332 qs_env=qs_env, sub_env=sub_env, &
333 wfm_rho_orb=work_matrices%rho_ao_orb_fm_sub, &
334 wfm_rho_aux_fit=work_matrices%rho_ao_aux_fit_fm_sub, &
335 wfm_aux_orb=work_matrices%wfm_aux_orb_sub)
338 CALL dbcsr_get_info(rho_ia_ao_aux_fit(1)%matrix, row_blk_size=blk_sizes)
339 ALLOCATE (a_xc_munu_sub(nspins))
341 ALLOCATE (a_xc_munu_sub(ispin)%matrix)
342 CALL dbcsr_create(matrix=a_xc_munu_sub(ispin)%matrix, name=
"ADMM_XC", &
343 dist=sub_env%dbcsr_dist, matrix_type=dbcsr_type_symmetric, &
344 row_blk_size=blk_sizes, col_blk_size=blk_sizes)
346 CALL dbcsr_set(a_xc_munu_sub(ispin)%matrix, 0.0_dp)
349 IF (admm_env%do_gapw)
THEN
350 basis_type =
"AUX_FIT_SOFT"
351 task_list => sub_env%task_list_aux_fit_soft
353 basis_type =
"AUX_FIT"
354 task_list => sub_env%task_list_aux_fit
357 IF (admm_env%do_gapw)
THEN
358 rho_atom_set => sub_env%local_rho_set_admm%rho_atom_set
359 rho1_atom_set => work_matrices%local_rho_set_admm%rho_atom_set
361 CALL qs_fxc_apply(qs_env, kernel_env_admm_aux%xc_deriv_set, kernel_env_admm_aux%xc_rho_set, &
362 work_matrices%rho_aux_fit_struct_sub, rho_atom_set, &
363 kernel_env_admm_aux%xc_section, admm_env%do_gapw, &
364 work_matrices%wpw_rspace_sub, fxc_tau, rho1_atom_set, &
365 do_scale=do_scale, is_triplet=is_rks_triplets, spinflip=do_noncol, &
366 pw_env_ext=sub_env%pw_env, kind_set_external=admm_env%admm_gapw_env%admm_kind_set, &
367 para_env_external=sub_env%para_env)
369 CALL pw_scale(work_matrices%wpw_rspace_sub(ispin), &
370 kernel_env_admm_aux%alpha*work_matrices%wpw_rspace_sub(ispin)%pw_grid%dvol)
371 CALL integrate_v_rspace(v_rspace=work_matrices%wpw_rspace_sub(ispin), &
372 hmat=a_xc_munu_sub(ispin), &
373 qs_env=qs_env, calculate_forces=.false., &
374 pw_env_external=sub_env%pw_env, &
375 basis_type=basis_type, &
376 task_list_external=task_list)
378 IF (admm_env%do_gapw)
THEN
379 CALL update_ks_atom(qs_env, a_xc_munu_sub, rho_ia_ao_aux_fit, forces=.false., tddft=.true., &
380 rho_atom_external=rho1_atom_set, &
381 kind_set_external=admm_env%admm_gapw_env%admm_kind_set, &
382 oce_external=admm_env%admm_gapw_env%oce, &
383 sab_external=sub_env%sab_aux_fit)
386 CALL dbcsr_create(dbwork, template=work_matrices%A_ia_munu_sub(1)%matrix)
388 matrix_struct=work_matrices%wfm_aux_orb_sub%matrix_struct)
390 matrix_struct=work_matrices%rho_ao_orb_fm_sub%matrix_struct)
391 CALL cp_fm_get_info(work_aux_orb, nrow_global=nao_aux, ncol_global=nao)
395 CALL parallel_gemm(
'T',
'N', nao, nao, nao_aux, 1.0_dp, sub_env%admm_A, &
396 work_aux_orb, 0.0_dp, work_orb_orb)
397 CALL dbcsr_copy(dbwork, work_matrices%A_ia_munu_sub(1)%matrix)
400 CALL dbcsr_add(work_matrices%A_ia_munu_sub(ispin)%matrix, dbwork, 1.0_dp, 1.0_dp)
409 DEALLOCATE (a_xc_munu_sub)
411 IF (admm_env%do_gapw)
THEN
412 cpwarn(
"GAPW/ADMM needs symmetric ADMM kernel")
413 cpabort(
"GAPW/ADMM@TDDFT")
415 CALL qs_fxc_apply(qs_env, kernel_env_admm_aux%xc_deriv_set, kernel_env_admm_aux%xc_rho_set, &
416 work_matrices%rho_aux_fit_struct_sub, rho_atom_set, &
417 kernel_env_admm_aux%xc_section, admm_env%do_gapw, &
418 work_matrices%wpw_rspace_sub, fxc_tau, rho1_atom_set, &
419 do_scale=do_scale, is_triplet=is_rks_triplets, spinflip=do_noncol, &
420 pw_env_ext=sub_env%pw_env)
423 CALL pw_axpy(work_matrices%wpw_rspace_sub(ispin), &
424 work_matrices%A_ia_rspace_sub(ispin), kernel_env_admm_aux%alpha)
433 IF (.NOT. dft_control%tddfpt2_control%do_bse_w_only)
THEN
434 IF (.NOT. dft_control%tddfpt2_control%do_bse_gw_only)
THEN
435 IF ((.NOT. is_rks_triplets) .AND. (spinflip ==
no_sf_tddfpt))
THEN
440 CALL pw_axpy(rho_ia_g(ispin), rho_ia_g(1))
443 rho_ia_g=rho_ia_g(1), &
444 local_rho_set=work_matrices%local_rho_set, &
445 hartree_local=work_matrices%hartree_local, &
446 qs_env=qs_env, sub_env=sub_env, gapw=gapw, &
447 work_v_gspace=work_matrices%wpw_gspace_sub(1), &
448 work_v_rspace=work_matrices%wpw_rspace_sub(1), &
449 tddfpt_mgrid=tddfpt_mgrid)
454 IF (.NOT. do_lrigpw)
THEN
455 CALL pw_scale(work_matrices%A_ia_rspace_sub(ispin), &
456 work_matrices%A_ia_rspace_sub(ispin)%pw_grid%dvol)
459 CALL integrate_v_rspace(v_rspace=work_matrices%A_ia_rspace_sub(ispin), &
460 hmat=work_matrices%A_ia_munu_sub(ispin), &
461 qs_env=qs_env, calculate_forces=.false., gapw=gapw, &
462 pw_env_external=sub_env%pw_env, &
463 task_list_external=sub_env%task_list_orb_soft)
464 ELSE IF (gapw_xc)
THEN
465 IF (.NOT. is_rks_triplets)
THEN
466 CALL integrate_v_rspace(v_rspace=work_matrices%A_ia_rspace_sub(ispin), &
467 hmat=work_matrices%A_ia_munu_sub(ispin), &
468 qs_env=qs_env, calculate_forces=.false., gapw=.false., &
469 pw_env_external=sub_env%pw_env, task_list_external=sub_env%task_list_orb)
472 CALL integrate_v_rspace(v_rspace=work_matrices%A_ia_rspace_sub(ispin), &
473 hmat=work_matrices%A_ia_munu_sub(ispin), &
474 qs_env=qs_env, calculate_forces=.false., gapw=.false., &
475 pw_env_external=sub_env%pw_env, task_list_external=sub_env%task_list_orb)
478 CALL pw_scale(work_matrices%A_ia_rspace_sub(ispin), &
479 work_matrices%A_ia_rspace_sub(ispin)%pw_grid%dvol)
480 lri_v_int => kernel_env%lri_density%lri_coefs(ispin)%lri_kinds
481 CALL get_qs_env(qs_env, nkind=nkind, para_env=para_env)
483 lri_v_int(ikind)%v_int = 0.0_dp
485 CALL integrate_v_rspace_one_center(work_matrices%A_ia_rspace_sub(ispin), &
486 qs_env, lri_v_int, .false.,
"P_LRI_AUX")
488 CALL para_env%sum(lri_v_int(ikind)%v_int)
494 IF (.NOT. do_lrigpw)
THEN
495 IF (gapw .OR. gapw_xc)
THEN
497 CALL update_ks_atom(qs_env, work_matrices%A_ia_munu_sub, rho_ia_ao, forces=.false., &
498 rho_atom_external=work_matrices%local_rho_set%rho_atom_set, &
505 IF (do_lrigpw .AND. (.NOT. is_rks_triplets))
THEN
506 CALL tddfpt2_lri_amat(qs_env, sub_env, kernel_env%lri_env, lri_v_int, work_matrices%A_ia_munu_sub)
510 IF (
ALLOCATED(work_matrices%evects_sub))
THEN
512 sub_env%mos_active(ispin), &
513 work_matrices%Aop_evects_sub(ispin, ivect), &
514 ncol=nactive(ispin), alpha=1.0_dp, beta=0.0_dp)
517 sub_env%mos_active(ispin), &
518 aop_evects(ispin, ivect), &
519 ncol=nactive(ispin), alpha=1.0_dp, beta=0.0_dp)
526 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.