96 do_hfx, do_admm, qs_env, kernel_env, kernel_env_admm_aux, &
97 sub_env, work_matrices, admm_symm, admm_xc_correction, do_lrigpw)
98 TYPE(
cp_fm_type),
DIMENSION(:, :),
INTENT(IN) :: aop_evects, evects
99 LOGICAL,
INTENT(in) :: is_rks_triplets, do_hfx, do_admm
104 LOGICAL,
INTENT(in) :: admm_symm, admm_xc_correction, do_lrigpw
106 CHARACTER(LEN=*),
PARAMETER :: routinen =
'fhxc_kernel'
108 CHARACTER(LEN=default_string_length) :: basis_type
109 INTEGER :: handle, ikind, ispin, ivect, nao, &
110 nao_aux, nkind, nspins, nvects
111 INTEGER,
DIMENSION(:),
POINTER :: blk_sizes
112 INTEGER,
DIMENSION(maxspins) :: nactive
113 LOGICAL :: gapw, gapw_xc
115 TYPE(
cp_fm_type) :: work_aux_orb, work_orb_orb
116 TYPE(dbcsr_p_type),
DIMENSION(:),
POINTER :: a_xc_munu_sub, rho_ia_ao, &
118 TYPE(dbcsr_type),
POINTER :: dbwork
122 TYPE(
pw_c1d_gs_type),
DIMENSION(:),
POINTER :: rho_ia_g, rho_ia_g_aux_fit
125 TYPE(
pw_r3d_rs_type),
DIMENSION(:),
POINTER :: rho_ia_r, rho_ia_r_aux_fit
126 TYPE(
rho_atom_type),
DIMENSION(:),
POINTER :: rho1_atom_set, rho_atom_set
129 CALL timeset(routinen, handle)
131 nspins =
SIZE(evects, 1)
132 nvects =
SIZE(evects, 2)
135 cpassert(
ASSOCIATED(sub_env%admm_A))
137 CALL get_qs_env(qs_env, dft_control=dft_control)
139 gapw = dft_control%qs_control%gapw
140 gapw_xc = dft_control%qs_control%gapw_xc
147 CALL qs_rho_get(work_matrices%rho_orb_struct_sub, rho_ao=rho_ia_ao, &
148 rho_g=rho_ia_g, rho_r=rho_ia_r)
149 IF (do_hfx .AND. do_admm)
THEN
151 CALL qs_rho_get(work_matrices%rho_aux_fit_struct_sub, &
152 rho_ao=rho_ia_ao_aux_fit, rho_g=rho_ia_g_aux_fit, &
153 rho_r=rho_ia_r_aux_fit)
157 IF (
ALLOCATED(work_matrices%evects_sub))
THEN
158 IF (
ASSOCIATED(work_matrices%evects_sub(1, ivect)%matrix_struct))
THEN
160 CALL dbcsr_set(rho_ia_ao(ispin)%matrix, 0.0_dp)
162 matrix_v=sub_env%mos_occ(ispin), &
163 matrix_g=work_matrices%evects_sub(ispin, ivect), &
164 ncol=nactive(ispin), symmetry_mode=1)
172 CALL dbcsr_set(rho_ia_ao(ispin)%matrix, 0.0_dp)
174 matrix_v=sub_env%mos_occ(ispin), &
175 matrix_g=evects(ispin, ivect), &
176 ncol=nactive(ispin), symmetry_mode=1)
182 pw_env_external=sub_env%pw_env, &
183 task_list_external=sub_env%task_list_orb, &
184 para_env_external=sub_env%para_env, &
185 tddfpt_lri_env=kernel_env%lri_env, &
186 tddfpt_lri_density=kernel_env%lri_density)
187 ELSEIF (dft_control%qs_control%lrigpw .OR. &
188 dft_control%qs_control%rigpw)
THEN
190 pw_env_external=sub_env%pw_env, &
191 task_list_external=sub_env%task_list_orb, &
192 para_env_external=sub_env%para_env)
196 local_rho_set=work_matrices%local_rho_set, &
197 pw_env_external=sub_env%pw_env, &
198 task_list_external=sub_env%task_list_orb_soft, &
199 para_env_external=sub_env%para_env)
201 do_rho0=(.NOT. is_rks_triplets))
202 ELSEIF (gapw_xc)
THEN
204 rho_xc_external=work_matrices%rho_xc_struct_sub, &
205 local_rho_set=work_matrices%local_rho_set, &
206 pw_env_external=sub_env%pw_env, &
207 task_list_external=sub_env%task_list_orb, &
208 task_list_external_soft=sub_env%task_list_orb_soft, &
209 para_env_external=sub_env%para_env)
213 pw_env_external=sub_env%pw_env, &
214 task_list_external=sub_env%task_list_orb, &
215 para_env_external=sub_env%para_env)
220 CALL dbcsr_set(work_matrices%A_ia_munu_sub(ispin)%matrix, 0.0_dp)
225 CALL pw_zero(work_matrices%A_ia_rspace_sub(ispin))
231 IF (kernel_env%do_exck)
THEN
234 CALL tddfpt_apply_xc(a_ia_rspace=work_matrices%A_ia_rspace_sub, kernel_env=kernel_env, &
235 rho_ia_struct=work_matrices%rho_xc_struct_sub, &
236 is_rks_triplets=is_rks_triplets, pw_env=sub_env%pw_env, &
237 work_v_xc=work_matrices%wpw_rspace_sub, &
238 work_v_xc_tau=work_matrices%wpw_tau_rspace_sub)
241 CALL pw_scale(work_matrices%A_ia_rspace_sub(ispin), &
242 work_matrices%A_ia_rspace_sub(ispin)%pw_grid%dvol)
243 CALL integrate_v_rspace(v_rspace=work_matrices%A_ia_rspace_sub(ispin), &
244 hmat=work_matrices%A_ia_munu_sub(ispin), &
245 qs_env=qs_env, calculate_forces=.false., gapw=gapw_xc, &
246 pw_env_external=sub_env%pw_env, &
247 task_list_external=sub_env%task_list_orb_soft)
248 CALL pw_zero(work_matrices%A_ia_rspace_sub(ispin))
251 IF (kernel_env%do_exck)
THEN
253 work_matrices%rho_orb_struct_sub, is_rks_triplets)
255 CALL tddfpt_apply_xc(a_ia_rspace=work_matrices%A_ia_rspace_sub, kernel_env=kernel_env, &
256 rho_ia_struct=work_matrices%rho_orb_struct_sub, &
257 is_rks_triplets=is_rks_triplets, pw_env=sub_env%pw_env, &
258 work_v_xc=work_matrices%wpw_rspace_sub, &
259 work_v_xc_tau=work_matrices%wpw_tau_rspace_sub)
262 IF (gapw .OR. gapw_xc)
THEN
263 rho_atom_set => sub_env%local_rho_set%rho_atom_set
264 rho1_atom_set => work_matrices%local_rho_set%rho_atom_set
266 sub_env%para_env, do_tddfpt2=.true., do_triplet=is_rks_triplets)
270 IF (do_admm .AND. admm_xc_correction)
THEN
273 rho_aux_fit_struct=work_matrices%rho_aux_fit_struct_sub, &
274 local_rho_set=work_matrices%local_rho_set_admm, &
275 qs_env=qs_env, sub_env=sub_env, &
276 wfm_rho_orb=work_matrices%rho_ao_orb_fm_sub, &
277 wfm_rho_aux_fit=work_matrices%rho_ao_aux_fit_fm_sub, &
278 wfm_aux_orb=work_matrices%wfm_aux_orb_sub)
281 CALL dbcsr_get_info(rho_ia_ao_aux_fit(1)%matrix, row_blk_size=blk_sizes)
282 ALLOCATE (a_xc_munu_sub(nspins))
284 ALLOCATE (a_xc_munu_sub(ispin)%matrix)
285 CALL dbcsr_create(matrix=a_xc_munu_sub(ispin)%matrix, name=
"ADMM_XC", &
286 dist=sub_env%dbcsr_dist, matrix_type=dbcsr_type_symmetric, &
287 row_blk_size=blk_sizes, col_blk_size=blk_sizes, nze=0)
289 CALL dbcsr_set(a_xc_munu_sub(ispin)%matrix, 0.0_dp)
292 CALL pw_env_get(sub_env%pw_env, auxbas_pw_pool=auxbas_pw_pool)
293 ALLOCATE (v_rspace_sub(nspins))
295 CALL auxbas_pw_pool%create_pw(v_rspace_sub(ispin))
296 CALL pw_zero(v_rspace_sub(ispin))
299 IF (admm_env%do_gapw)
THEN
300 basis_type =
"AUX_FIT_SOFT"
301 task_list => sub_env%task_list_aux_fit_soft
303 basis_type =
"AUX_FIT"
304 task_list => sub_env%task_list_aux_fit
308 kernel_env=kernel_env_admm_aux, &
309 rho_ia_struct=work_matrices%rho_aux_fit_struct_sub, &
310 is_rks_triplets=is_rks_triplets, pw_env=sub_env%pw_env, &
311 work_v_xc=work_matrices%wpw_rspace_sub, &
312 work_v_xc_tau=work_matrices%wpw_tau_rspace_sub)
314 CALL pw_scale(v_rspace_sub(ispin), v_rspace_sub(ispin)%pw_grid%dvol)
315 CALL integrate_v_rspace(v_rspace=v_rspace_sub(ispin), &
316 hmat=a_xc_munu_sub(ispin), &
317 qs_env=qs_env, calculate_forces=.false., &
318 pw_env_external=sub_env%pw_env, &
319 basis_type=basis_type, &
320 task_list_external=task_list)
322 IF (admm_env%do_gapw)
THEN
323 rho_atom_set => sub_env%local_rho_set_admm%rho_atom_set
324 rho1_atom_set => work_matrices%local_rho_set_admm%rho_atom_set
326 kernel_env_admm_aux%xc_section, &
327 sub_env%para_env, do_tddfpt2=.true., &
328 do_triplet=is_rks_triplets, &
329 kind_set_external=admm_env%admm_gapw_env%admm_kind_set)
330 CALL update_ks_atom(qs_env, a_xc_munu_sub, rho_ia_ao_aux_fit, forces=.false., tddft=.true., &
331 rho_atom_external=rho1_atom_set, &
332 kind_set_external=admm_env%admm_gapw_env%admm_kind_set, &
333 oce_external=admm_env%admm_gapw_env%oce, &
334 sab_external=sub_env%sab_aux_fit)
337 CALL dbcsr_create(dbwork, template=work_matrices%A_ia_munu_sub(1)%matrix)
339 matrix_struct=work_matrices%wfm_aux_orb_sub%matrix_struct)
341 matrix_struct=work_matrices%rho_ao_orb_fm_sub%matrix_struct)
342 CALL cp_fm_get_info(work_aux_orb, nrow_global=nao_aux, ncol_global=nao)
346 CALL parallel_gemm(
'T',
'N', nao, nao, nao_aux, 1.0_dp, sub_env%admm_A, &
347 work_aux_orb, 0.0_dp, work_orb_orb)
348 CALL dbcsr_copy(dbwork, work_matrices%A_ia_munu_sub(1)%matrix)
349 CALL dbcsr_set(dbwork, 0.0_dp)
351 CALL dbcsr_add(work_matrices%A_ia_munu_sub(ispin)%matrix, dbwork, 1.0_dp, 1.0_dp)
353 CALL dbcsr_release(dbwork)
356 CALL auxbas_pw_pool%give_back_pw(v_rspace_sub(ispin))
358 DEALLOCATE (v_rspace_sub)
362 CALL dbcsr_deallocate_matrix(a_xc_munu_sub(ispin)%matrix)
364 DEALLOCATE (a_xc_munu_sub)
367 kernel_env=kernel_env_admm_aux, &
368 rho_ia_struct=work_matrices%rho_aux_fit_struct_sub, &
369 is_rks_triplets=is_rks_triplets, pw_env=sub_env%pw_env, &
370 work_v_xc=work_matrices%wpw_rspace_sub, &
371 work_v_xc_tau=work_matrices%wpw_tau_rspace_sub)
372 IF (admm_env%do_gapw)
THEN
373 cpwarn(
"GAPW/ADMM needs symmetric ADMM kernel")
374 cpabort(
"GAPW/ADMM@TDDFT")
381 IF (.NOT. is_rks_triplets)
THEN
386 CALL pw_axpy(rho_ia_g(ispin), rho_ia_g(1))
389 rho_ia_g=rho_ia_g(1), &
390 local_rho_set=work_matrices%local_rho_set, &
391 hartree_local=work_matrices%hartree_local, &
392 qs_env=qs_env, sub_env=sub_env, gapw=gapw, &
393 work_v_gspace=work_matrices%wpw_gspace_sub(1), &
394 work_v_rspace=work_matrices%wpw_rspace_sub(1))
399 IF (.NOT. do_lrigpw)
THEN
400 CALL pw_scale(work_matrices%A_ia_rspace_sub(ispin), &
401 work_matrices%A_ia_rspace_sub(ispin)%pw_grid%dvol)
404 CALL integrate_v_rspace(v_rspace=work_matrices%A_ia_rspace_sub(ispin), &
405 hmat=work_matrices%A_ia_munu_sub(ispin), &
406 qs_env=qs_env, calculate_forces=.false., gapw=gapw, &
407 pw_env_external=sub_env%pw_env, &
408 task_list_external=sub_env%task_list_orb_soft)
409 ELSEIF (gapw_xc)
THEN
410 IF (.NOT. is_rks_triplets)
THEN
411 CALL integrate_v_rspace(v_rspace=work_matrices%A_ia_rspace_sub(ispin), &
412 hmat=work_matrices%A_ia_munu_sub(ispin), &
413 qs_env=qs_env, calculate_forces=.false., gapw=.false., &
414 pw_env_external=sub_env%pw_env, task_list_external=sub_env%task_list_orb)
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 pw_scale(work_matrices%A_ia_rspace_sub(ispin), &
424 work_matrices%A_ia_rspace_sub(ispin)%pw_grid%dvol)
425 lri_v_int => kernel_env%lri_density%lri_coefs(ispin)%lri_kinds
426 CALL get_qs_env(qs_env, nkind=nkind, para_env=para_env)
428 lri_v_int(ikind)%v_int = 0.0_dp
430 CALL integrate_v_rspace_one_center(work_matrices%A_ia_rspace_sub(ispin), &
431 qs_env, lri_v_int, .false.,
"P_LRI_AUX")
433 CALL para_env%sum(lri_v_int(ikind)%v_int)
439 IF (.NOT. do_lrigpw)
THEN
440 IF (gapw .OR. gapw_xc)
THEN
442 CALL update_ks_atom(qs_env, work_matrices%A_ia_munu_sub, rho_ia_ao, forces=.false., &
443 rho_atom_external=work_matrices%local_rho_set%rho_atom_set, &
450 IF (do_lrigpw .AND. (.NOT. is_rks_triplets))
THEN
451 CALL tddfpt2_lri_amat(qs_env, sub_env, kernel_env%lri_env, lri_v_int, work_matrices%A_ia_munu_sub)
454 IF (
ALLOCATED(work_matrices%evects_sub))
THEN
457 sub_env%mos_occ(ispin), &
458 work_matrices%Aop_evects_sub(ispin, ivect), &
459 ncol=nactive(ispin), alpha=1.0_dp, beta=0.0_dp)
464 sub_env%mos_occ(ispin), &
465 aop_evects(ispin, ivect), &
466 ncol=nactive(ispin), alpha=1.0_dp, beta=0.0_dp)
471 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_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, 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, rhs)
Get the QUICKSTEP environment.