GPU Force Halo Exchange

[alexxy/gromacs.git] / src / gromacs / nbnxm / nbnxm.h

diff --git a/src/gromacs/nbnxm/nbnxm.h b/src/gromacs/nbnxm/nbnxm.h
index 7a27a5cb4e10af0696e90c2d5dc87b30577c89c1..caa21d9c7f0a4a72e4e09ee14db334c64a354fd3 100644 (file)
--- a/src/gromacs/nbnxm/nbnxm.h
+++ b/src/gromacs/nbnxm/nbnxm.h
@@ -406,6 +406,12 @@ struct nonbonded_verlet_t
         /*! \brief Wait for non-local copy of coordinate buffer from device to host */
         void wait_nonlocal_x_copy_D2H_done();
 
+        /*! \brief return GPU pointer to f in rvec format */
+        void* get_gpu_frvec();
+
+        /*! \brief Ensure local stream waits for non-local stream */
+        void stream_local_wait_for_nonlocal();
+
         //! Return the kernel setup
         const Nbnxm::KernelSetup &kernelSetup() const
         {