Remove excessive H2D and D2H coordinates copies when update is offloaded

[alexxy/gromacs.git] / src / gromacs / mdtypes / state_propagator_data_gpu.h

diff --git a/src/gromacs/mdtypes/state_propagator_data_gpu.h b/src/gromacs/mdtypes/state_propagator_data_gpu.h
index 96066d0ac3f948cd0b5320de6276a10d9b10be81..b5bc2699c0bcbae12af7f45e46e17a6919de237b 100644 (file)
--- a/src/gromacs/mdtypes/state_propagator_data_gpu.h
+++ b/src/gromacs/mdtypes/state_propagator_data_gpu.h
@@ -215,6 +215,14 @@ class StatePropagatorDataGpu
                                                                const SimulationWorkload &simulationWork,
                                                                const StepWorkload       &stepWork);
 
+        /*! \brief Blocking wait until coordinates are copied to the device.
+         *
+         * Synchronizes the stream in which the copy was executed.
+         *
+         *  \param[in] atomLocality  Locality of the particles to wait for.
+         */
+        void waitCoordinatesCopiedToDevice(AtomLocality  atomLocality);
+
         /*! \brief Getter for the event synchronizer for the update is done on th GPU
          *
          *  \returns  The event to synchronize the stream coordinates wre updated on device.