fix compile errors

Signed-off-by: Jinzhe Zeng <[email protected]>
deepmodeling · Sep 19, 2024 · 9381a13 · 9381a13
1 parent 1422e2b
commit 9381a13
Showing 1 changed file with 9 additions and 9 deletions.
diff --git a/source/api_cc/tests/test_deeppot_dpa_pt.cc b/source/api_cc/tests/test_deeppot_dpa_pt.cc
@@ -171,7 +171,7 @@ class TestInferDeepPotDpaPtNopbc : public ::testing::Test {
                                   00.25, 3.32, 1.68, 3.36,  3.00, 1.81,
                                   3.51,  2.51, 2.60, 4.27,  3.22, 1.56};
   std::vector<int> atype = {0, 1, 1, 0, 1, 1};
-  std::vector<VALUETYPE> box = nullptr;
+  std::vector<VALUETYPE> box = {};
   // Generated by the following Python code:
   // import numpy as np
   // from deepmd.infer import DeepPot
@@ -329,11 +329,11 @@ TYPED_TEST(TestInferDeepPotDpaPtNopbc, cpu_lmp_nlist) {
   std::vector<std::vector<int> > nlist_data = {
       {1, 2, 3, 4, 5}, {0, 2, 3, 4, 5}, {0, 1, 3, 4, 5},
       {0, 1, 2, 4, 5}, {0, 1, 2, 3, 5}, {0, 1, 2, 3, 4}};
-  std::vector<int> ilist(nloc), numneigh(nloc);
-  std::vector<int*> firstneigh(nloc);
-  deepmd::InputNlist inlist(nloc, &ilist[0], &numneigh[0], &firstneigh[0]);
+  std::vector<int> ilist(natoms), numneigh(natoms);
+  std::vector<int*> firstneigh(natoms);
+  deepmd::InputNlist inlist(natoms, &ilist[0], &numneigh[0], &firstneigh[0]);
   convert_nlist(inlist, nlist_data);
-  dp.compute(ener, force, virial, coord, atype, box, 0, nlist, 0);
+  dp.compute(ener, force, virial, coord, atype, box, 0, inlist, 0);
 
   EXPECT_EQ(force.size(), natoms * 3);
   EXPECT_EQ(virial.size(), 9);
@@ -365,12 +365,12 @@ TYPED_TEST(TestInferDeepPotDpaPtNopbc, cpu_lmp_nlist_atomic) {
   std::vector<std::vector<int> > nlist_data = {
       {1, 2, 3, 4, 5}, {0, 2, 3, 4, 5}, {0, 1, 3, 4, 5},
       {0, 1, 2, 4, 5}, {0, 1, 2, 3, 5}, {0, 1, 2, 3, 4}};
-  std::vector<int> ilist(nloc), numneigh(nloc);
-  std::vector<int*> firstneigh(nloc);
-  deepmd::InputNlist inlist(nloc, &ilist[0], &numneigh[0], &firstneigh[0]);
+  std::vector<int> ilist(natoms), numneigh(natoms);
+  std::vector<int*> firstneigh(natoms);
+  deepmd::InputNlist inlist(natoms, &ilist[0], &numneigh[0], &firstneigh[0]);
   convert_nlist(inlist, nlist_data);
   dp.compute(ener, force, virial, atom_ener, atom_vir, coord, atype, box, 0,
-             nlist, 0);
+             inlist, 0);
 
   EXPECT_EQ(force.size(), natoms * 3);
   EXPECT_EQ(virial.size(), 9);