Pass cFREEZE and cTC to update after DD partitioning

[alexxy/gromacs.git] / src / gromacs / mdlib / tests / leapfrogtestdata.cpp

diff --git a/src/gromacs/mdlib/tests/leapfrogtestdata.cpp b/src/gromacs/mdlib/tests/leapfrogtestdata.cpp
index c00fbbd4c001e0aa4bae2419286fddcf215a931f..9d1f7780be68a05a173877aad803ecd15881f374 100644 (file)
--- a/src/gromacs/mdlib/tests/leapfrogtestdata.cpp
+++ b/src/gromacs/mdlib/tests/leapfrogtestdata.cpp
@@ -172,7 +172,9 @@ LeapFrogTestData::LeapFrogTestData(int        numAtoms,
     mdAtoms_.cFREEZE                  = nullptr;
 
     update_ = std::make_unique<Update>(inputRecord_, nullptr);
-    update_->setNumAtoms(numAtoms);
+    update_->updateAfterPartition(numAtoms,
+                                  gmx::ArrayRef<const unsigned short>(),
+                                  gmx::arrayRefFromArray(mdAtoms_.cTC, mdAtoms_.nr));
 
     doPressureCouple_ = (nstpcouple != 0);