Skip to content

Commit f42ee4d

Browse files
committed
Add UT for cal_gdmepsl and modify the ref files to suit the new data structure.
1 parent 865f3da commit f42ee4d

25 files changed

+2215
-1018
lines changed

source/module_hamilt_lcao/module_deepks/test/LCAO_deepks_test.cpp

Lines changed: 26 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -196,6 +196,32 @@ void test_deepks::check_gdmx(torch::Tensor& gdmx)
196196
}
197197
}
198198

199+
void test_deepks::check_gdmepsl()
200+
{
201+
torch::Tensor gdmepsl;
202+
if (PARAM.sys.gamma_only_local)
203+
{
204+
GlobalC::ld.cal_gdmepsl(dm_new, ucell, ORB, Test_Deepks::GridD, kv.nkstot, kv.kvec_d, GlobalC::ld.phialpha, gdmepsl);
205+
}
206+
else
207+
{
208+
GlobalC::ld
209+
.cal_gdmx(dm_k_new, ucell, ORB, Test_Deepks::GridD, kv.nkstot, kv.kvec_d, GlobalC::ld.phialpha, gdmepsl);
210+
}
211+
GlobalC::ld.check_gdmepsl(gdmepsl);
212+
213+
for (int i = 0; i < 6; i++)
214+
{
215+
std::stringstream ss;
216+
std::stringstream ss1;
217+
ss.str("");
218+
ss << "gdmepsl_" << i << ".dat";
219+
ss1.str("");
220+
ss1 << "gdmepsl_" << i << "_ref.dat";
221+
this->compare_with_ref(ss.str(), ss1.str());
222+
}
223+
}
224+
199225
void test_deepks::check_descriptor(std::vector<torch::Tensor>& descriptor)
200226
{
201227
DeePKS_domain::cal_descriptor(ucell.nat,

source/module_hamilt_lcao/module_deepks/test/LCAO_deepks_test.h

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -98,6 +98,7 @@ class test_deepks
9898

9999
void check_pdm();
100100
void check_gdmx(torch::Tensor& gdmx);
101+
void check_gdmepsl();
101102

102103
void check_descriptor(std::vector<torch::Tensor>& descriptor);
103104
void check_gvx(torch::Tensor& gdmx);

source/module_hamilt_lcao/module_deepks/test/Makefile.Objects

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,6 @@ HEADERS= *.h
77

88
OBJS_MAIN=klist_1.o\
99
parallel_orbitals.o\
10-
nnr.o\
1110
LCAO_deepks.o\
1211
LCAO_deepks_fdelta.o\
1312
LCAO_deepks_io.o\

source/module_hamilt_lcao/module_deepks/test/main_deepks.cpp

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -38,6 +38,7 @@ int calculate()
3838

3939
torch::Tensor gdmx;
4040
test.check_gdmx(gdmx);
41+
test.check_gdmepsl();
4142

4243
std::vector<torch::Tensor> descriptor;
4344
test.check_descriptor(descriptor);

tests/deepks/604_NO_deepks_ut_CH4_gamma/gdmepsl_0_ref.dat

Lines changed: 195 additions & 0 deletions
Large diffs are not rendered by default.

tests/deepks/604_NO_deepks_ut_CH4_gamma/gdmepsl_1_ref.dat

Lines changed: 195 additions & 0 deletions
Large diffs are not rendered by default.

tests/deepks/604_NO_deepks_ut_CH4_gamma/gdmepsl_2_ref.dat

Lines changed: 195 additions & 0 deletions
Large diffs are not rendered by default.

tests/deepks/604_NO_deepks_ut_CH4_gamma/gdmepsl_3_ref.dat

Lines changed: 195 additions & 0 deletions
Large diffs are not rendered by default.

tests/deepks/604_NO_deepks_ut_CH4_gamma/gdmepsl_4_ref.dat

Lines changed: 195 additions & 0 deletions
Large diffs are not rendered by default.

tests/deepks/604_NO_deepks_ut_CH4_gamma/gdmepsl_5_ref.dat

Lines changed: 195 additions & 0 deletions
Large diffs are not rendered by default.

0 commit comments

Comments
 (0)