41#include "./base/base_uses.f90"
61 CHARACTER(len=*),
PARAMETER :: routinen =
'pao_param_init_gth'
63 INTEGER :: acol, arow, handle, iatom,
idx, ikind, &
64 iterm, jatom, maxl, n, natoms
65 INTEGER,
DIMENSION(:),
POINTER :: blk_sizes_pri, col_blk_size, nterms, &
67 REAL(
dp),
DIMENSION(:, :),
POINTER :: block_v_term, vec_v_terms
74 CALL timeset(routinen, handle)
79 qs_kind_set=qs_kind_set, &
80 particle_set=particle_set)
83 ALLOCATE (row_blk_size(natoms), col_blk_size(natoms), nterms(natoms))
85 CALL get_atomic_kind(particle_set(iatom)%atomic_kind, kind_number=ikind)
94 CALL dbcsr_get_info(matrix_s(1)%matrix, row_blk_size=blk_sizes_pri)
95 col_blk_size = sum(nterms)
96 row_blk_size = blk_sizes_pri**2
98 name=
"PAO matrix_V_terms", &
99 dist=pao%diag_distribution, &
101 row_blk_size=row_blk_size, &
102 col_blk_size=col_blk_size)
104 CALL dbcsr_set(pao%matrix_V_terms, 0.0_dp)
112 iatom = arow; cpassert(arow == acol)
113 n = blk_sizes_pri(iatom)
115 IF (jatom == iatom) cycle
116 DO iterm = 1, nterms(jatom)
117 idx = sum(nterms(1:jatom - 1)) + iterm
118 block_v_term(1:n, 1:n) => vec_v_terms(:,
idx)
119 CALL gth_calc_term(qs_env, block_v_term, iatom, jatom, iterm)
126 IF (pao%precondition)
THEN
127 CALL pao_param_gth_preconditioner(pao, qs_env, nterms)
130 DEALLOCATE (row_blk_size, col_blk_size, nterms)
131 CALL timestop(handle)
142 IF (pao%precondition)
THEN
155 SUBROUTINE pao_param_gth_preconditioner(pao, qs_env, nterms)
158 INTEGER,
DIMENSION(:),
POINTER :: nterms
160 CHARACTER(len=*),
PARAMETER :: routinen =
'pao_param_gth_preconditioner'
162 INTEGER :: acol, arow, handle, i, iatom, ioffset, &
163 j, jatom, joffset, m, n, natoms
164 LOGICAL :: arnoldi_converged, converged, found
165 REAL(
dp) :: eval_max, eval_min
166 REAL(
dp),
DIMENSION(:, :),
POINTER :: block, block_overlap, block_v_term
172 CALL timeset(routinen, handle)
174 CALL get_qs_env(qs_env, ls_scf_env=ls_scf_env)
176 natoms =
SIZE(nterms)
179 template=pao%matrix_V_terms, &
181 row_blk_size=nterms, &
184 CALL dbcsr_set(matrix_gth_overlap, 0.0_dp)
188 ioffset = sum(nterms(1:iatom - 1))
189 joffset = sum(nterms(1:jatom - 1))
193 ALLOCATE (block(n, m))
200 cpassert(arow == acol)
203 block(i, j) = block(i, j) + sum(block_v_term(:, ioffset + i)*block_v_term(:, joffset + j))
209 CALL group%sum(block)
211 CALL dbcsr_get_block_p(matrix=matrix_gth_overlap, row=iatom, col=jatom, block=block_overlap, found=found)
212 IF (
ASSOCIATED(block_overlap))
THEN
213 block_overlap = block
221 CALL arnoldi_extremal(matrix_gth_overlap, eval_max, eval_min, max_iter=100, &
222 threshold=1e-2_dp, converged=arnoldi_converged)
223 IF (pao%iw > 0)
WRITE (pao%iw, *)
"PAO| GTH-preconditioner converged, min, max, max/min:", &
224 arnoldi_converged, eval_min, eval_max, eval_max/eval_min
226 CALL dbcsr_create(pao%matrix_precon, template=matrix_gth_overlap)
227 CALL dbcsr_create(pao%matrix_precon_inv, template=matrix_gth_overlap)
230 threshold=ls_scf_env%eps_filter, &
231 order=ls_scf_env%s_sqrt_order, &
232 max_iter_lanczos=ls_scf_env%max_iter_lanczos, &
233 eps_lanczos=ls_scf_env%eps_lanczos, &
237 IF (.NOT. converged)
THEN
238 cpabort(
"PAO: Sqrt of GTH-preconditioner did not converge.")
241 CALL timestop(handle)
242 END SUBROUTINE pao_param_gth_preconditioner
256 LOGICAL,
INTENT(IN) :: gradient
257 REAL(
dp),
INTENT(INOUT),
OPTIONAL :: penalty
259 CHARACTER(len=*),
PARAMETER :: routinen =
'pao_calc_AB_gth'
265 CALL timeset(routinen, handle)
267 CALL dbcsr_create(matrix_u, matrix_type=
"N", dist=pao%diag_distribution, template=matrix_s(1)%matrix)
273 CALL pao_calc_u_gth(pao, matrix_u, matrix_m, pao%matrix_G, penalty)
276 CALL pao_calc_u_gth(pao, matrix_u, penalty=penalty)
281 CALL timestop(handle)
292 SUBROUTINE pao_calc_u_gth(pao, matrix_U, matrix_M1, matrix_G, penalty)
295 TYPE(
dbcsr_type),
OPTIONAL :: matrix_m1, matrix_g
296 REAL(
dp),
INTENT(INOUT),
OPTIONAL :: penalty
298 CHARACTER(len=*),
PARAMETER :: routinen =
'pao_calc_U_gth'
300 INTEGER :: acol, arow, handle, iatom,
idx, iterm, &
302 INTEGER,
DIMENSION(:),
POINTER :: nterms
304 REAL(
dp),
ALLOCATABLE,
DIMENSION(:) :: gaps
305 REAL(
dp),
DIMENSION(:),
POINTER :: world_g, world_x
306 REAL(
dp),
DIMENSION(:, :),
POINTER :: block_g, block_m1, block_m2, block_u, &
307 block_v, block_v_term, block_x, &
312 CALL timeset(routinen, handle)
314 CALL dbcsr_get_info(pao%matrix_X, row_blk_size=nterms, group=group)
315 natoms =
SIZE(nterms)
316 ALLOCATE (gaps(natoms))
320 ALLOCATE (world_x(sum(nterms)), world_g(sum(nterms)))
321 world_x = 0.0_dp; world_g = 0.0_dp
327 iatom = arow; cpassert(arow == acol)
328 idx = sum(nterms(1:iatom - 1))
329 world_x(
idx + 1:
idx + nterms(iatom)) = block_x(:, 1)
332 CALL group%sum(world_x)
338 iatom = arow; cpassert(arow == acol)
340 CALL dbcsr_get_block_p(matrix=pao%matrix_V_terms, row=iatom, col=iatom, block=vec_v_terms, found=found)
341 cpassert(
ASSOCIATED(vec_v_terms))
344 ALLOCATE (block_v(n, n))
346 DO iterm = 1,
SIZE(world_x)
347 block_v_term(1:n, 1:n) => vec_v_terms(:, iterm)
348 block_v = block_v + world_x(iterm)*block_v_term
352 IF (.NOT.
PRESENT(matrix_g))
THEN
356 cpassert(
PRESENT(matrix_m1))
357 CALL dbcsr_get_block_p(matrix=matrix_m1, row=iatom, col=iatom, block=block_m1, found=found)
358 ALLOCATE (block_m2(n, n))
360 m1=block_m1, g=block_m2, gap=gaps(iatom))
361 DO iterm = 1,
SIZE(world_g)
362 block_v_term(1:n, 1:n) => vec_v_terms(:, iterm)
363 world_g(iterm) = world_g(iterm) + sum(block_v_term*block_m2)
365 DEALLOCATE (block_m2)
372 IF (
PRESENT(matrix_g))
THEN
373 CALL group%sum(world_g)
377 iatom = arow; cpassert(arow == acol)
378 idx = sum(nterms(1:iatom - 1))
379 block_g(:, 1) = world_g(
idx + 1:
idx + nterms(iatom))
384 DEALLOCATE (world_x, world_g)
387 IF (
PRESENT(penalty))
THEN
388 CALL group%sum(penalty)
393 IF (pao%iw_gap > 0)
THEN
395 WRITE (pao%iw_gap, *)
"PAO| atom:", iatom,
" fock gap:", gaps(iatom)
402 WRITE (pao%iw,
"(A,E20.10,A,T71,I10)")
" PAO| min_gap:", minval(gaps),
" for atom:", minloc(gaps)
406 CALL timestop(handle)
408 END SUBROUTINE pao_calc_u_gth
418 INTEGER,
INTENT(IN) :: ikind
419 INTEGER,
INTENT(OUT) :: nparams
421 INTEGER :: max_projector, maxl, ncombis
423 TYPE(
qs_kind_type),
DIMENSION(:),
POINTER :: qs_kind_set
425 CALL get_qs_env(qs_env, qs_kind_set=qs_kind_set)
429 cpabort(
"GTH parametrization requires exactly one PAO_POTENTIAL section per KIND")
436 cpabort(
"GTH parametrization requires non-negative PAO_POTENTIAL%MAXL")
439 IF (max_projector < 0)
THEN
440 cpabort(
"GTH parametrization requires non-negative PAO_POTENTIAL%MAX_PROJECTOR")
443 IF (mod(maxl, 2) /= 0)
THEN
444 cpabort(
"GTH parametrization requires even-numbered PAO_POTENTIAL%MAXL")
447 ncombis = (max_projector + 1)*(max_projector + 2)/2
448 nparams = ncombis*(maxl/2 + 1)
460 SUBROUTINE gth_calc_term(qs_env, block_V, iatom, jatom, kterm)
462 REAL(
dp),
DIMENSION(:, :),
INTENT(OUT) :: block_v
463 INTEGER,
INTENT(IN) :: iatom, jatom, kterm
465 INTEGER :: c, ikind, jkind, lpot, max_l, min_l, &
466 pot_max_projector, pot_maxl
467 REAL(
dp),
DIMENSION(3) :: ra, rab, rb
468 REAL(kind=
dp) :: pot_beta
473 TYPE(
qs_kind_type),
DIMENSION(:),
POINTER :: qs_kind_set
477 particle_set=particle_set, &
478 qs_kind_set=qs_kind_set)
481 CALL get_atomic_kind(particle_set(jatom)%atomic_kind, kind_number=jkind)
489 outer:
DO lpot = 0, pot_maxl, 2
490 DO max_l = 0, pot_max_projector
493 IF (c == kterm)
EXIT outer
499 CALL get_atomic_kind(particle_set(iatom)%atomic_kind, kind_number=ikind)
500 CALL get_qs_kind(qs_kind_set(ikind), basis_set=basis_set)
502 ra = particle_set(iatom)%r
503 rb = particle_set(jatom)%r
504 rab =
pbc(ra, rb, cell)
508 min_l=min_l, max_l=max_l, beta=pot_beta, weight=1.0_dp)
510 END SUBROUTINE gth_calc_term
519 INTEGER :: acol, arow
520 REAL(
dp),
DIMENSION(:, :),
POINTER :: block_x
528 cpassert(arow == acol)
529 cpassert(
SIZE(block_x, 2) == 1)
533 block_x(1, 1) = 0.01_dp
static GRID_HOST_DEVICE int idx(const orbital a)
Return coset index of given orbital angular momentum.
arnoldi iteration using dbcsr
subroutine, public arnoldi_extremal(matrix_a, max_ev, min_ev, converged, threshold, max_iter)
simple wrapper to estimate extremal eigenvalues with arnoldi, using the old lanczos interface this hi...
Define the atomic kind types and their sub types.
subroutine, public get_atomic_kind(atomic_kind, fist_potential, element_symbol, name, mass, kind_number, natom, atom_list, rcov, rvdw, z, qeff, apol, cpol, mm_radius, shell, shell_active, damping)
Get attributes of an atomic kind.
Handles all functions related to the CELL.
logical function, public dbcsr_iterator_blocks_left(iterator)
...
subroutine, public dbcsr_iterator_stop(iterator)
...
subroutine, public dbcsr_get_block_p(matrix, row, col, block, found, row_size, col_size)
...
subroutine, public dbcsr_get_info(matrix, nblkrows_total, nblkcols_total, nfullrows_total, nfullcols_total, nblkrows_local, nblkcols_local, nfullrows_local, nfullcols_local, my_prow, my_pcol, local_rows, local_cols, proc_row_dist, proc_col_dist, row_blk_size, col_blk_size, row_blk_offset, col_blk_offset, distribution, name, matrix_type, group)
...
subroutine, public dbcsr_iterator_next_block(iterator, row, column, block, block_number_argument_has_been_removed, row_size, col_size, row_offset, col_offset, transposed)
...
subroutine, public dbcsr_iterator_start(iterator, matrix, shared, dynamic, dynamic_byrows)
...
subroutine, public dbcsr_set(matrix, alpha)
...
subroutine, public dbcsr_release(matrix)
...
subroutine, public dbcsr_reserve_all_blocks(matrix)
Reserves all blocks.
subroutine, public dbcsr_reserve_diag_blocks(matrix)
Reserves all diagonal blocks.
Types needed for a linear scaling quickstep SCF run based on the density matrix.
Routines useful for iterative matrix calculations.
subroutine, public matrix_sqrt_newton_schulz(matrix_sqrt, matrix_sqrt_inv, matrix, threshold, order, eps_lanczos, max_iter_lanczos, symmetrize, converged, iounit)
compute the sqrt of a matrix via the sign function and the corresponding Newton-Schulz iterations the...
Defines the basic variable types.
integer, parameter, public dp
Machine interface based on Fortran 2003 and POSIX.
subroutine, public m_flush(lunit)
flushes units if the &GLOBAL flag is set accordingly
Interface to the message passing library MPI.
Provides Cartesian and spherical orbital pointers and indices.
subroutine, public init_orbital_pointers(maxl)
Initialize or update the orbital pointers.
Common framework for using eigenvectors of a Fock matrix as PAO basis.
subroutine, public pao_calc_u_block_fock(pao, iatom, v, u, penalty, gap, evals, m1, g)
Calculate new matrix U and optinally its gradient G.
Parametrization based on GTH pseudo potentials.
subroutine, public pao_calc_ab_gth(pao, qs_env, ls_scf_env, gradient, penalty)
Takes current matrix_X and calculates the matrices A and B.
subroutine, public pao_param_initguess_gth(pao)
Calculate initial guess for matrix_X.
subroutine, public pao_param_count_gth(qs_env, ikind, nparams)
Returns the number of parameters for given atomic kind.
subroutine, public pao_param_init_gth(pao, qs_env)
Initialize the linear potential parametrization.
subroutine, public pao_param_finalize_gth(pao)
Finalize the GTH potential parametrization.
Common routines for PAO parametrizations.
subroutine, public pao_calc_grad_lnv_wrt_u(qs_env, ls_scf_env, matrix_m_diag)
Helper routine, calculates partial derivative dE/dU.
subroutine, public pao_calc_ab_from_u(pao, qs_env, ls_scf_env, matrix_u_diag)
Takes current matrix_X and calculates the matrices A and B.
Factory routines for potentials used e.g. by pao_param_exp and pao_ml.
subroutine, public pao_calc_gaussian(basis_set, block_v, block_d, rab, lpot, beta, weight, min_shell, max_shell, min_l, max_l)
Calculates potential term of the form r**lpot * Exp(-beta*r**2) One needs to call init_orbital_pointe...
Types used by the PAO machinery.
Define the data structure for the particle information.
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, 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.
Define the quickstep kind type and their sub types.
subroutine, public get_qs_kind(qs_kind, basis_set, basis_type, ncgf, nsgf, all_potential, tnadd_potential, gth_potential, sgp_potential, upf_potential, cneo_potential, se_parameter, dftb_parameter, xtb_parameter, dftb3_param, zatom, zeff, elec_conf, mao, lmax_dftb, alpha_core_charge, ccore_charge, core_charge, core_charge_radius, paw_proj_set, paw_atom, hard_radius, hard0_radius, max_rad_local, covalent_radius, vdw_radius, gpw_type_forced, harmonics, max_iso_not0, max_s_harm, grid_atom, ngrid_ang, ngrid_rad, lmax_rho0, dft_plus_u_atom, l_of_dft_plus_u, n_of_dft_plus_u, u_minus_j, u_of_dft_plus_u, j_of_dft_plus_u, alpha_of_dft_plus_u, beta_of_dft_plus_u, j0_of_dft_plus_u, occupation_of_dft_plus_u, dispersion, bs_occupation, magnetization, no_optimize, addel, laddel, naddel, orbitals, max_scf, eps_scf, smear, u_ramping, u_minus_j_target, eps_u_ramping, init_u_ramping_each_scf, reltmat, ghost, monovalent, floating, name, element_symbol, pao_basis_size, pao_model_file, pao_potentials, pao_descriptors, nelec)
Get attributes of an atomic kind.
Type defining parameters related to the simulation cell.
Holds information about a PAO potential.
Provides all information about a quickstep kind.