Use RVec instead of float for x, v and f device buffers

[alexxy/gromacs.git] / src / gromacs / mdlib / update_constrain_gpu_impl.cu

diff --git a/src/gromacs/mdlib/update_constrain_gpu_impl.cu b/src/gromacs/mdlib/update_constrain_gpu_impl.cu
index ebbe15a02ff5059be0305a748f05651ad9fe4a33..6991ef0dc351ca2732eb0e3e46c3c68eadd959fc 100644 (file)
--- a/src/gromacs/mdlib/update_constrain_gpu_impl.cu
+++ b/src/gromacs/mdlib/update_constrain_gpu_impl.cu
@@ -188,12 +188,12 @@ UpdateConstrainGpu::Impl::Impl(const t_inputrec&     ir,
 
 UpdateConstrainGpu::Impl::~Impl() {}
 
-void UpdateConstrainGpu::Impl::set(DeviceBuffer<float>       d_x,
-                                   DeviceBuffer<float>       d_v,
-                                   const DeviceBuffer<float> d_f,
-                                   const t_idef&             idef,
-                                   const t_mdatoms&          md,
-                                   const int                 numTempScaleValues)
+void UpdateConstrainGpu::Impl::set(DeviceBuffer<RVec>       d_x,
+                                   DeviceBuffer<RVec>       d_v,
+                                   const DeviceBuffer<RVec> d_f,
+                                   const t_idef&            idef,
+                                   const t_mdatoms&         md,
+                                   const int                numTempScaleValues)
 {
     GMX_ASSERT(d_x != nullptr, "Coordinates device buffer should not be null.");
     GMX_ASSERT(d_v != nullptr, "Velocities device buffer should not be null.");
@@ -259,12 +259,12 @@ void UpdateConstrainGpu::scaleCoordinates(const matrix scalingMatrix)
     impl_->scaleCoordinates(scalingMatrix);
 }
 
-void UpdateConstrainGpu::set(DeviceBuffer<float>       d_x,
-                             DeviceBuffer<float>       d_v,
-                             const DeviceBuffer<float> d_f,
-                             const t_idef&             idef,
-                             const t_mdatoms&          md,
-                             const int                 numTempScaleValues)
+void UpdateConstrainGpu::set(DeviceBuffer<RVec>       d_x,
+                             DeviceBuffer<RVec>       d_v,
+                             const DeviceBuffer<RVec> d_f,
+                             const t_idef&            idef,
+                             const t_mdatoms&         md,
+                             const int                numTempScaleValues)
 {
     impl_->set(d_x, d_v, d_f, idef, md, numTempScaleValues);
 }