97 do_hfx, do_admm, qs_env, kernel_env, kernel_env_admm_aux, &
98 sub_env, work_matrices, admm_symm, admm_xc_correction, do_lrigpw, &
100 TYPE(
cp_fm_type),
DIMENSION(:, :),
INTENT(INOUT) :: aop_evects
101 TYPE(
cp_fm_type),
DIMENSION(:, :),
INTENT(IN) :: evects
102 LOGICAL,
INTENT(in) :: is_rks_triplets, do_hfx, do_admm
107 LOGICAL,
INTENT(in) :: admm_symm, admm_xc_correction, &
108 do_lrigpw, tddfpt_mgrid
110 CHARACTER(LEN=*),
PARAMETER :: routinen =
'fhxc_kernel'
112 CHARACTER(LEN=default_string_length) :: basis_type
113 INTEGER :: handle, ikind, ispin, ivect, nao, &
114 nao_aux, nkind, nspins, nvects
115 INTEGER,
DIMENSION(:),
POINTER :: blk_sizes
116 INTEGER,
DIMENSION(maxspins) :: nactive
117 LOGICAL :: gapw, gapw_xc
119 TYPE(
cp_fm_type) :: work_aux_orb, work_orb_orb
120 TYPE(
dbcsr_p_type),
DIMENSION(:),
POINTER :: a_xc_munu_sub, rho_ia_ao, &
126 TYPE(
pw_c1d_gs_type),
DIMENSION(:),
POINTER :: rho_ia_g, rho_ia_g_aux_fit
129 TYPE(
pw_r3d_rs_type),
DIMENSION(:),
POINTER :: rho_ia_r, rho_ia_r_aux_fit
130 TYPE(
rho_atom_type),
DIMENSION(:),
POINTER :: rho1_atom_set, rho_atom_set
133 CALL timeset(routinen, handle)
135 nspins =
SIZE(evects, 1)
136 nvects =
SIZE(evects, 2)
139 cpassert(
ASSOCIATED(sub_env%admm_A))
141 CALL get_qs_env(qs_env, dft_control=dft_control)
143 gapw = dft_control%qs_control%gapw
144 gapw_xc = dft_control%qs_control%gapw_xc
151 CALL qs_rho_get(work_matrices%rho_orb_struct_sub, rho_ao=rho_ia_ao, &
152 rho_g=rho_ia_g, rho_r=rho_ia_r)
153 IF (do_hfx .AND. do_admm)
THEN
155 CALL qs_rho_get(work_matrices%rho_aux_fit_struct_sub, &
156 rho_ao=rho_ia_ao_aux_fit, rho_g=rho_ia_g_aux_fit, &
157 rho_r=rho_ia_r_aux_fit)
161 IF (
ALLOCATED(work_matrices%evects_sub))
THEN
162 IF (
ASSOCIATED(work_matrices%evects_sub(1, ivect)%matrix_struct))
THEN
164 CALL dbcsr_set(rho_ia_ao(ispin)%matrix, 0.0_dp)
166 matrix_v=sub_env%mos_occ(ispin), &
167 matrix_g=work_matrices%evects_sub(ispin, ivect), &
168 ncol=nactive(ispin), symmetry_mode=1)
176 CALL dbcsr_set(rho_ia_ao(ispin)%matrix, 0.0_dp)
178 matrix_v=sub_env%mos_occ(ispin), &
179 matrix_g=evects(ispin, ivect), &
180 ncol=nactive(ispin), symmetry_mode=1)
186 pw_env_external=sub_env%pw_env, &
187 task_list_external=sub_env%task_list_orb, &
188 para_env_external=sub_env%para_env, &
189 tddfpt_lri_env=kernel_env%lri_env, &
190 tddfpt_lri_density=kernel_env%lri_density)
191 ELSEIF (dft_control%qs_control%lrigpw .OR. &
192 dft_control%qs_control%rigpw)
THEN
194 pw_env_external=sub_env%pw_env, &
195 task_list_external=sub_env%task_list_orb, &
196 para_env_external=sub_env%para_env)
200 local_rho_set=work_matrices%local_rho_set, &
201 pw_env_external=sub_env%pw_env, &
202 task_list_external=sub_env%task_list_orb_soft, &
203 para_env_external=sub_env%para_env)
205 do_rho0=(.NOT. is_rks_triplets), pw_env_sub=sub_env%pw_env)
206 ELSEIF (gapw_xc)
THEN
208 rho_xc_external=work_matrices%rho_xc_struct_sub, &
209 local_rho_set=work_matrices%local_rho_set, &
210 pw_env_external=sub_env%pw_env, &
211 task_list_external=sub_env%task_list_orb, &
212 task_list_external_soft=sub_env%task_list_orb_soft, &
213 para_env_external=sub_env%para_env)
214 CALL prepare_gapw_den(qs_env, work_matrices%local_rho_set, do_rho0=.false., &
215 pw_env_sub=sub_env%pw_env)
218 pw_env_external=sub_env%pw_env, &
219 task_list_external=sub_env%task_list_orb, &
220 para_env_external=sub_env%para_env)
225 CALL dbcsr_set(work_matrices%A_ia_munu_sub(ispin)%matrix, 0.0_dp)
230 CALL pw_zero(work_matrices%A_ia_rspace_sub(ispin))
236 IF (kernel_env%do_exck)
THEN
239 CALL tddfpt_apply_xc(a_ia_rspace=work_matrices%A_ia_rspace_sub, kernel_env=kernel_env, &
240 rho_ia_struct=work_matrices%rho_xc_struct_sub, &
241 is_rks_triplets=is_rks_triplets, pw_env=sub_env%pw_env, &
242 work_v_xc=work_matrices%wpw_rspace_sub, &
243 work_v_xc_tau=work_matrices%wpw_tau_rspace_sub)
246 CALL pw_scale(work_matrices%A_ia_rspace_sub(ispin), &
247 work_matrices%A_ia_rspace_sub(ispin)%pw_grid%dvol)
248 CALL integrate_v_rspace(v_rspace=work_matrices%A_ia_rspace_sub(ispin), &
249 hmat=work_matrices%A_ia_munu_sub(ispin), &
250 qs_env=qs_env, calculate_forces=.false., gapw=gapw_xc, &
251 pw_env_external=sub_env%pw_env, &
252 task_list_external=sub_env%task_list_orb_soft)
253 CALL pw_zero(work_matrices%A_ia_rspace_sub(ispin))
256 IF (kernel_env%do_exck)
THEN
258 work_matrices%rho_orb_struct_sub, is_rks_triplets)
260 CALL tddfpt_apply_xc(a_ia_rspace=work_matrices%A_ia_rspace_sub, kernel_env=kernel_env, &
261 rho_ia_struct=work_matrices%rho_orb_struct_sub, &
262 is_rks_triplets=is_rks_triplets, pw_env=sub_env%pw_env, &
263 work_v_xc=work_matrices%wpw_rspace_sub, &
264 work_v_xc_tau=work_matrices%wpw_tau_rspace_sub)
267 IF (gapw .OR. gapw_xc)
THEN
268 rho_atom_set => sub_env%local_rho_set%rho_atom_set
269 rho1_atom_set => work_matrices%local_rho_set%rho_atom_set
271 sub_env%para_env, do_tddfpt2=.true., do_triplet=is_rks_triplets)
275 IF (do_admm .AND. admm_xc_correction)
THEN
278 rho_aux_fit_struct=work_matrices%rho_aux_fit_struct_sub, &
279 local_rho_set=work_matrices%local_rho_set_admm, &
280 qs_env=qs_env, sub_env=sub_env, &
281 wfm_rho_orb=work_matrices%rho_ao_orb_fm_sub, &
282 wfm_rho_aux_fit=work_matrices%rho_ao_aux_fit_fm_sub, &
283 wfm_aux_orb=work_matrices%wfm_aux_orb_sub)
286 CALL dbcsr_get_info(rho_ia_ao_aux_fit(1)%matrix, row_blk_size=blk_sizes)
287 ALLOCATE (a_xc_munu_sub(nspins))
289 ALLOCATE (a_xc_munu_sub(ispin)%matrix)
290 CALL dbcsr_create(matrix=a_xc_munu_sub(ispin)%matrix, name=
"ADMM_XC", &
291 dist=sub_env%dbcsr_dist, matrix_type=dbcsr_type_symmetric, &
292 row_blk_size=blk_sizes, col_blk_size=blk_sizes)
294 CALL dbcsr_set(a_xc_munu_sub(ispin)%matrix, 0.0_dp)
297 CALL pw_env_get(sub_env%pw_env, auxbas_pw_pool=auxbas_pw_pool)
298 ALLOCATE (v_rspace_sub(nspins))
300 CALL auxbas_pw_pool%create_pw(v_rspace_sub(ispin))
301 CALL pw_zero(v_rspace_sub(ispin))
304 IF (admm_env%do_gapw)
THEN
305 basis_type =
"AUX_FIT_SOFT"
306 task_list => sub_env%task_list_aux_fit_soft
308 basis_type =
"AUX_FIT"
309 task_list => sub_env%task_list_aux_fit
313 kernel_env=kernel_env_admm_aux, &
314 rho_ia_struct=work_matrices%rho_aux_fit_struct_sub, &
315 is_rks_triplets=is_rks_triplets, pw_env=sub_env%pw_env, &
316 work_v_xc=work_matrices%wpw_rspace_sub, &
317 work_v_xc_tau=work_matrices%wpw_tau_rspace_sub)
319 CALL pw_scale(v_rspace_sub(ispin), v_rspace_sub(ispin)%pw_grid%dvol)
320 CALL integrate_v_rspace(v_rspace=v_rspace_sub(ispin), &
321 hmat=a_xc_munu_sub(ispin), &
322 qs_env=qs_env, calculate_forces=.false., &
323 pw_env_external=sub_env%pw_env, &
324 basis_type=basis_type, &
325 task_list_external=task_list)
327 IF (admm_env%do_gapw)
THEN
328 rho_atom_set => sub_env%local_rho_set_admm%rho_atom_set
329 rho1_atom_set => work_matrices%local_rho_set_admm%rho_atom_set
331 kernel_env_admm_aux%xc_section, &
332 sub_env%para_env, do_tddfpt2=.true., &
333 do_triplet=is_rks_triplets, &
334 kind_set_external=admm_env%admm_gapw_env%admm_kind_set)
335 CALL update_ks_atom(qs_env, a_xc_munu_sub, rho_ia_ao_aux_fit, forces=.false., tddft=.true., &
336 rho_atom_external=rho1_atom_set, &
337 kind_set_external=admm_env%admm_gapw_env%admm_kind_set, &
338 oce_external=admm_env%admm_gapw_env%oce, &
339 sab_external=sub_env%sab_aux_fit)
342 CALL dbcsr_create(dbwork, template=work_matrices%A_ia_munu_sub(1)%matrix)
344 matrix_struct=work_matrices%wfm_aux_orb_sub%matrix_struct)
346 matrix_struct=work_matrices%rho_ao_orb_fm_sub%matrix_struct)
347 CALL cp_fm_get_info(work_aux_orb, nrow_global=nao_aux, ncol_global=nao)
351 CALL parallel_gemm(
'T',
'N', nao, nao, nao_aux, 1.0_dp, sub_env%admm_A, &
352 work_aux_orb, 0.0_dp, work_orb_orb)
353 CALL dbcsr_copy(dbwork, work_matrices%A_ia_munu_sub(1)%matrix)
356 CALL dbcsr_add(work_matrices%A_ia_munu_sub(ispin)%matrix, dbwork, 1.0_dp, 1.0_dp)
361 CALL auxbas_pw_pool%give_back_pw(v_rspace_sub(ispin))
363 DEALLOCATE (v_rspace_sub)
369 DEALLOCATE (a_xc_munu_sub)
372 kernel_env=kernel_env_admm_aux, &
373 rho_ia_struct=work_matrices%rho_aux_fit_struct_sub, &
374 is_rks_triplets=is_rks_triplets, pw_env=sub_env%pw_env, &
375 work_v_xc=work_matrices%wpw_rspace_sub, &
376 work_v_xc_tau=work_matrices%wpw_tau_rspace_sub)
377 IF (admm_env%do_gapw)
THEN
378 cpwarn(
"GAPW/ADMM needs symmetric ADMM kernel")
379 cpabort(
"GAPW/ADMM@TDDFT")
386 IF (.NOT. is_rks_triplets)
THEN
391 CALL pw_axpy(rho_ia_g(ispin), rho_ia_g(1))
394 rho_ia_g=rho_ia_g(1), &
395 local_rho_set=work_matrices%local_rho_set, &
396 hartree_local=work_matrices%hartree_local, &
397 qs_env=qs_env, sub_env=sub_env, gapw=gapw, &
398 work_v_gspace=work_matrices%wpw_gspace_sub(1), &
399 work_v_rspace=work_matrices%wpw_rspace_sub(1), &
400 tddfpt_mgrid=tddfpt_mgrid)
405 IF (.NOT. do_lrigpw)
THEN
406 CALL pw_scale(work_matrices%A_ia_rspace_sub(ispin), &
407 work_matrices%A_ia_rspace_sub(ispin)%pw_grid%dvol)
410 CALL integrate_v_rspace(v_rspace=work_matrices%A_ia_rspace_sub(ispin), &
411 hmat=work_matrices%A_ia_munu_sub(ispin), &
412 qs_env=qs_env, calculate_forces=.false., gapw=gapw, &
413 pw_env_external=sub_env%pw_env, &
414 task_list_external=sub_env%task_list_orb_soft)
415 ELSEIF (gapw_xc)
THEN
416 IF (.NOT. is_rks_triplets)
THEN
417 CALL integrate_v_rspace(v_rspace=work_matrices%A_ia_rspace_sub(ispin), &
418 hmat=work_matrices%A_ia_munu_sub(ispin), &
419 qs_env=qs_env, calculate_forces=.false., gapw=.false., &
420 pw_env_external=sub_env%pw_env, task_list_external=sub_env%task_list_orb)
423 CALL integrate_v_rspace(v_rspace=work_matrices%A_ia_rspace_sub(ispin), &
424 hmat=work_matrices%A_ia_munu_sub(ispin), &
425 qs_env=qs_env, calculate_forces=.false., gapw=.false., &
426 pw_env_external=sub_env%pw_env, task_list_external=sub_env%task_list_orb)
429 CALL pw_scale(work_matrices%A_ia_rspace_sub(ispin), &
430 work_matrices%A_ia_rspace_sub(ispin)%pw_grid%dvol)
431 lri_v_int => kernel_env%lri_density%lri_coefs(ispin)%lri_kinds
432 CALL get_qs_env(qs_env, nkind=nkind, para_env=para_env)
434 lri_v_int(ikind)%v_int = 0.0_dp
436 CALL integrate_v_rspace_one_center(work_matrices%A_ia_rspace_sub(ispin), &
437 qs_env, lri_v_int, .false.,
"P_LRI_AUX")
439 CALL para_env%sum(lri_v_int(ikind)%v_int)
445 IF (.NOT. do_lrigpw)
THEN
446 IF (gapw .OR. gapw_xc)
THEN
448 CALL update_ks_atom(qs_env, work_matrices%A_ia_munu_sub, rho_ia_ao, forces=.false., &
449 rho_atom_external=work_matrices%local_rho_set%rho_atom_set, &
456 IF (do_lrigpw .AND. (.NOT. is_rks_triplets))
THEN
457 CALL tddfpt2_lri_amat(qs_env, sub_env, kernel_env%lri_env, lri_v_int, work_matrices%A_ia_munu_sub)
460 IF (
ALLOCATED(work_matrices%evects_sub))
THEN
463 sub_env%mos_occ(ispin), &
464 work_matrices%Aop_evects_sub(ispin, ivect), &
465 ncol=nactive(ispin), alpha=1.0_dp, beta=0.0_dp)
470 sub_env%mos_occ(ispin), &
471 aop_evects(ispin, ivect), &
472 ncol=nactive(ispin), alpha=1.0_dp, beta=0.0_dp)
477 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, 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, 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, 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, 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, ecoul_1c, rho0_s_rs, rho0_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)
Get the QUICKSTEP environment.