Use DeviceBuffer<RVec> in GPU force reduction and PME code

[alexxy/gromacs.git] / src / gromacs / ewald / pme_force_sender_gpu.h

diff --git a/src/gromacs/ewald/pme_force_sender_gpu.h b/src/gromacs/ewald/pme_force_sender_gpu.h
index 081ba454e63ae688f05f470d7f7f4781e05970ba..261260977993847c232c726f5778007944890a1d 100644 (file)
--- a/src/gromacs/ewald/pme_force_sender_gpu.h
+++ b/src/gromacs/ewald/pme_force_sender_gpu.h
@@ -45,6 +45,7 @@
 #include <memory>
 
 #include "gromacs/math/vectypes.h"
+#include "gromacs/gpu_utils/devicebuffer_datatype.h"
 #include "gromacs/utility/gmxmpi.h"
 
 class GpuEventSynchronizer;
@@ -83,7 +84,7 @@ public:
      * Initialization of GPU PME Force sender
      * \param[in] d_f   force buffer in GPU memory
      */
-    void sendForceBufferAddressToPpRanks(rvec* d_f);
+    void sendForceBufferAddressToPpRanks(DeviceBuffer<RVec> d_f);
 
     /*! \brief
      * Send force synchronizer to PP rank