Make use of the DeviceStreamManager

[alexxy/gromacs.git] / src / gromacs / gpu_utils / device_stream_manager.h

diff --git a/src/gromacs/gpu_utils/device_stream_manager.h b/src/gromacs/gpu_utils/device_stream_manager.h
index 4cfa6161a429c46d1426c6dd65338cedc931904b..4565d1ac0c2044633420536cd4aecfffea5e07d6 100644 (file)
--- a/src/gromacs/gpu_utils/device_stream_manager.h
+++ b/src/gromacs/gpu_utils/device_stream_manager.h
@@ -124,6 +124,12 @@ public:
      */
     const DeviceStream& stream(DeviceStreamType streamToGet) const;
 
+    /*! \brief Returns a handle to the GPU stream to compute bonded forces in.
+     *
+     * \param[in] hasPPDomainDecomposition Whether there is a particle-particle domain decomposition.
+     */
+    const DeviceStream& bondedStream(bool hasPPDomainDecomposition) const;
+
     /*! \brief Return whether the requested GPU stream is valid for use.
      *
      * \param[in] streamToCheck Which stream to check.