forked from cp2k/cp2k
-
Notifications
You must be signed in to change notification settings - Fork 0
/
admm_utils.F
178 lines (141 loc) · 7.84 KB
/
admm_utils.F
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
!--------------------------------------------------------------------------------------------------!
! CP2K: A general program to perform molecular dynamics simulations !
! Copyright 2000-2023 CP2K developers group <https://cp2k.org> !
! !
! SPDX-License-Identifier: GPL-2.0-or-later !
!--------------------------------------------------------------------------------------------------!
! **************************************************************************************************
!> \brief Contains methods used in the context of density fitting
!> \par History
!> 04.2008 created [Manuel Guidon]
!> 02.2013 moved from admm_methods
!> \author Manuel Guidon
! **************************************************************************************************
MODULE admm_utils
USE admm_types, ONLY: admm_type
USE cp_dbcsr_operations, ONLY: copy_fm_to_dbcsr
USE dbcsr_api, ONLY: dbcsr_add,&
dbcsr_copy,&
dbcsr_create,&
dbcsr_deallocate_matrix,&
dbcsr_set,&
dbcsr_type,&
dbcsr_type_symmetric
USE input_constants, ONLY: do_admm_purify_cauchy,&
do_admm_purify_cauchy_subspace,&
do_admm_purify_mo_diag,&
do_admm_purify_mo_no_diag,&
do_admm_purify_none
USE kinds, ONLY: dp
USE parallel_gemm_api, ONLY: parallel_gemm
#include "./base/base_uses.f90"
IMPLICIT NONE
PRIVATE
PUBLIC :: admm_correct_for_eigenvalues, &
admm_uncorrect_for_eigenvalues
CHARACTER(len=*), PARAMETER, PRIVATE :: moduleN = 'admm_utils'
!***
CONTAINS
! **************************************************************************************************
!> \brief ...
!> \param ispin ...
!> \param admm_env ...
!> \param ks_matrix ...
! **************************************************************************************************
SUBROUTINE admm_correct_for_eigenvalues(ispin, admm_env, ks_matrix)
INTEGER, INTENT(IN) :: ispin
TYPE(admm_type), POINTER :: admm_env
TYPE(dbcsr_type), POINTER :: ks_matrix
INTEGER :: nao_aux_fit, nao_orb
TYPE(dbcsr_type), POINTER :: work
nao_aux_fit = admm_env%nao_aux_fit
nao_orb = admm_env%nao_orb
IF (.NOT. admm_env%block_dm) THEN
SELECT CASE (admm_env%purification_method)
CASE (do_admm_purify_cauchy_subspace)
!* remove what has been added and add the correction
NULLIFY (work)
ALLOCATE (work)
CALL dbcsr_create(work, template=ks_matrix, name='work', matrix_type=dbcsr_type_symmetric)
CALL dbcsr_copy(work, ks_matrix)
CALL dbcsr_set(work, 0.0_dp)
CALL copy_fm_to_dbcsr(admm_env%ks_to_be_merged(ispin), work, keep_sparsity=.TRUE.)
CALL dbcsr_add(ks_matrix, work, 1.0_dp, -1.0_dp)
! ** calculate A^T*H_tilde*A
CALL parallel_gemm('N', 'N', nao_aux_fit, nao_orb, nao_aux_fit, &
1.0_dp, admm_env%K(ispin), admm_env%A, 0.0_dp, &
admm_env%work_aux_orb)
CALL parallel_gemm('T', 'N', nao_orb, nao_orb, nao_aux_fit, &
1.0_dp, admm_env%A, admm_env%work_aux_orb, 0.0_dp, &
admm_env%H_corr(ispin))
CALL copy_fm_to_dbcsr(admm_env%H_corr(ispin), work, keep_sparsity=.TRUE.)
CALL dbcsr_add(ks_matrix, work, 1.0_dp, 1.0_dp)
CALL dbcsr_deallocate_matrix(work)
CASE (do_admm_purify_mo_diag)
!* remove what has been added and add the correction
NULLIFY (work)
ALLOCATE (work)
CALL dbcsr_create(work, template=ks_matrix, name='work', matrix_type=dbcsr_type_symmetric)
CALL dbcsr_copy(work, ks_matrix)
CALL dbcsr_set(work, 0.0_dp)
CALL copy_fm_to_dbcsr(admm_env%ks_to_be_merged(ispin), work, keep_sparsity=.TRUE.)
! ** calculate A^T*H_tilde*A
CALL parallel_gemm('N', 'N', nao_aux_fit, nao_orb, nao_aux_fit, &
1.0_dp, admm_env%K(ispin), admm_env%A, 0.0_dp, &
admm_env%work_aux_orb)
CALL parallel_gemm('T', 'N', nao_orb, nao_orb, nao_aux_fit, &
1.0_dp, admm_env%A, admm_env%work_aux_orb, 0.0_dp, &
admm_env%H_corr(ispin))
CALL copy_fm_to_dbcsr(admm_env%H_corr(ispin), work, keep_sparsity=.TRUE.)
CALL dbcsr_add(ks_matrix, work, 1.0_dp, 1.0_dp)
CALL dbcsr_deallocate_matrix(work)
CASE (do_admm_purify_mo_no_diag, do_admm_purify_none, do_admm_purify_cauchy)
! do nothing
END SELECT
END IF
END SUBROUTINE admm_correct_for_eigenvalues
! **************************************************************************************************
!> \brief ...
!> \param ispin ...
!> \param admm_env ...
!> \param ks_matrix ...
! **************************************************************************************************
SUBROUTINE admm_uncorrect_for_eigenvalues(ispin, admm_env, ks_matrix)
INTEGER, INTENT(IN) :: ispin
TYPE(admm_type), POINTER :: admm_env
TYPE(dbcsr_type), POINTER :: ks_matrix
INTEGER :: nao_aux_fit, nao_orb
TYPE(dbcsr_type), POINTER :: work
nao_aux_fit = admm_env%nao_aux_fit
nao_orb = admm_env%nao_orb
IF (.NOT. admm_env%block_dm) THEN
SELECT CASE (admm_env%purification_method)
CASE (do_admm_purify_cauchy_subspace)
!* remove what has been added and add the correction
NULLIFY (work)
ALLOCATE (work)
CALL dbcsr_create(work, template=ks_matrix, name='work', matrix_type=dbcsr_type_symmetric)
CALL dbcsr_copy(work, ks_matrix)
CALL dbcsr_set(work, 0.0_dp)
CALL copy_fm_to_dbcsr(admm_env%H_corr(ispin), work, keep_sparsity=.TRUE.)
CALL dbcsr_add(ks_matrix, work, 1.0_dp, -1.0_dp)
CALL copy_fm_to_dbcsr(admm_env%H_corr(ispin), work, keep_sparsity=.TRUE.)
CALL dbcsr_set(work, 0.0_dp)
CALL copy_fm_to_dbcsr(admm_env%ks_to_be_merged(ispin), work, keep_sparsity=.TRUE.)
CALL dbcsr_add(ks_matrix, work, 1.0_dp, 1.0_dp)
CALL dbcsr_deallocate_matrix(work)
CASE (do_admm_purify_mo_diag)
NULLIFY (work)
ALLOCATE (work)
CALL dbcsr_create(work, template=ks_matrix, name='work', matrix_type=dbcsr_type_symmetric)
CALL dbcsr_copy(work, ks_matrix)
CALL dbcsr_set(work, 0.0_dp)
CALL copy_fm_to_dbcsr(admm_env%H_corr(ispin), work, keep_sparsity=.TRUE.)
CALL dbcsr_add(ks_matrix, work, 1.0_dp, -1.0_dp)
CALL dbcsr_deallocate_matrix(work)
CASE (do_admm_purify_mo_no_diag, do_admm_purify_none, do_admm_purify_cauchy)
! do nothing
END SELECT
END IF
END SUBROUTINE admm_uncorrect_for_eigenvalues
END MODULE admm_utils