Pass the GPU streams to StatePropagatorDataGpu constructor

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

diff --git a/src/gromacs/mdlib/update_constrain_cuda_impl.cu b/src/gromacs/mdlib/update_constrain_cuda_impl.cu
index b22f365358da3a593ae10f53b233dd89e7fb4ea1..e4667720947eb63fc93ce77a5f3442495a435558 100644 (file)
--- a/src/gromacs/mdlib/update_constrain_cuda_impl.cu
+++ b/src/gromacs/mdlib/update_constrain_cuda_impl.cu
@@ -162,6 +162,11 @@ void UpdateConstrainCuda::Impl::setPbc(const t_pbc *pbc)
     settleCuda_->setPbc(pbc);
 }
 
+void UpdateConstrainCuda::Impl::synchronizeStream()
+{
+    gpuStreamSynchronize(commandStream_);
+}
+
 UpdateConstrainCuda::UpdateConstrainCuda(const t_inputrec  &ir,
                                          const gmx_mtop_t  &mtop,
                                          const void        *commandStream)
@@ -201,4 +206,9 @@ void UpdateConstrainCuda::setPbc(const t_pbc *pbc)
     impl_->setPbc(pbc);
 }
 
+void UpdateConstrainCuda::synchronizeStream()
+{
+    impl_->synchronizeStream();
+}
+
 } //namespace gmx