Skip to content

Commit d0bedae

Browse files
authored
Add files via upload
1 parent e27ac49 commit d0bedae

File tree

1 file changed

+12
-8
lines changed

1 file changed

+12
-8
lines changed

source/module_esolver/cal_edm_tddft.cpp

Lines changed: 12 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,7 @@
22

33
#include "module_io/cal_r_overlap_R.h"
44
#include "module_io/dipole_io.h"
5+
#include "module_io/rho_io.h"
56
#include "module_io/td_current_io.h"
67
#include "module_io/write_HS.h"
78
#include "module_io/write_HS_R.h"
@@ -41,13 +42,16 @@ namespace ModuleESolver
4142
void ESolver_KS_LCAO_TDDFT::cal_edm_tddft()
4243
{
4344
// mohan add 2024-03-27
44-
const int nlocal = PARAM.globalv.nlocal;
45+
const int nlocal = GlobalV::NLOCAL;
4546
assert(nlocal >= 0);
4647

4748
dynamic_cast<elecstate::ElecStateLCAO<std::complex<double>>*>(this->pelec)
4849
->get_DM()
4950
->EDMK.resize(kv.get_nks());
5051
for (int ik = 0; ik < kv.get_nks(); ++ik) {
52+
53+
p_hamilt->updateHk(ik);
54+
5155
std::complex<double>* tmp_dmk
5256
= dynamic_cast<elecstate::ElecStateLCAO<std::complex<double>>*>(this->pelec)->get_DM()->get_DMK_pointer(ik);
5357

@@ -146,11 +150,11 @@ void ESolver_KS_LCAO_TDDFT::cal_edm_tddft()
146150
&nlocal,
147151
&nlocal,
148152
&one_float,
149-
tmp_dmk,
153+
Htmp,
150154
&one_int,
151155
&one_int,
152156
this->pv.desc,
153-
Htmp,
157+
Sinv,
154158
&one_int,
155159
&one_int,
156160
this->pv.desc,
@@ -160,7 +164,7 @@ void ESolver_KS_LCAO_TDDFT::cal_edm_tddft()
160164
&one_int,
161165
this->pv.desc);
162166

163-
pzgemm_(&N_char,
167+
pzgemm_(&T_char,
164168
&N_char,
165169
&nlocal,
166170
&nlocal,
@@ -170,7 +174,7 @@ void ESolver_KS_LCAO_TDDFT::cal_edm_tddft()
170174
&one_int,
171175
&one_int,
172176
this->pv.desc,
173-
Sinv,
177+
tmp_dmk,
174178
&one_int,
175179
&one_int,
176180
this->pv.desc,
@@ -201,16 +205,16 @@ void ESolver_KS_LCAO_TDDFT::cal_edm_tddft()
201205
this->pv.desc);
202206

203207
pzgemm_(&N_char,
204-
&N_char,
208+
&T_char,
205209
&nlocal,
206210
&nlocal,
207211
&nlocal,
208212
&one_float,
209-
tmp3,
213+
tmp_dmk,
210214
&one_int,
211215
&one_int,
212216
this->pv.desc,
213-
tmp_dmk,
217+
tmp3,
214218
&one_int,
215219
&one_int,
216220
this->pv.desc,

0 commit comments

Comments
 (0)