Support pinning in HostAllocator

[alexxy/gromacs.git] / src / gromacs / mdlib / mdatoms.h

diff --git a/src/gromacs/mdlib/mdatoms.h b/src/gromacs/mdlib/mdatoms.h
index b537e0df2f6b25c54124752394f05a5d026e14f3..8e708b3e0dde55f4cead60c7ae9b1101d44f26a4 100644 (file)
--- a/src/gromacs/mdlib/mdatoms.h
+++ b/src/gromacs/mdlib/mdatoms.h
@@ -69,11 +69,11 @@ class MDAtoms
     //! C-style mdatoms struct.
     unique_cptr<t_mdatoms> mdatoms_;
     //! Memory for chargeA that can be set up for efficient GPU transfer.
-    std::vector < real, HostAllocator < real>> chargeA_;
+    HostVector<real>       chargeA_;
     public:
         // TODO make this private
         //! Constructor.
-        MDAtoms(HostAllocationPolicy policy);
+        MDAtoms();
         //! Getter.
         t_mdatoms *mdatoms()
         {