Skip to content

Commit 60c50b1

Browse files
author
Han Wang
committed
fill the force and atom energy/virial of ghost atoms with 0s
1 parent 4d86ff2 commit 60c50b1

File tree

1 file changed

+16
-4
lines changed

1 file changed

+16
-4
lines changed

source/lib/src/NNPInter.cc

Lines changed: 16 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -295,7 +295,11 @@ run_model (ENERGYTYPE & dener,
295295
unsigned nall = nloc + nghost;
296296
if (nloc == 0) {
297297
dener = 0;
298-
dforce_.clear();
298+
// no backward map needed
299+
// dforce of size nall * 3
300+
dforce_.resize(nall * 3);
301+
fill(dforce_.begin(), dforce_.end(), 0.0);
302+
// dvirial of size 9
299303
dvirial.resize(9);
300304
fill(dvirial.begin(), dvirial.end(), 0.0);
301305
return;
@@ -344,11 +348,19 @@ run_model (ENERGYTYPE & dener,
344348
unsigned nall = nloc + nghost;
345349
if (nloc == 0) {
346350
dener = 0;
347-
dforce_.clear();
351+
// no backward map needed
352+
// dforce of size nall * 3
353+
dforce_.resize(nall * 3);
354+
fill(dforce_.begin(), dforce_.end(), 0.0);
355+
// dvirial of size 9
348356
dvirial.resize(9);
349357
fill(dvirial.begin(), dvirial.end(), 0.0);
350-
datom_energy_.clear();
351-
datom_virial_.clear();
358+
// datom_energy_ of size nall
359+
datom_energy_.resize(nall);
360+
fill(datom_energy_.begin(), datom_energy_.end(), 0.0);
361+
// datom_virial_ of size nall * 9
362+
datom_virial_.resize(nall * 9);
363+
fill(datom_virial_.begin(), datom_virial_.end(), 0.0);
352364
return;
353365
}
354366

0 commit comments

Comments
 (0)