Fix cycle counters for "comm.coord" and "Wait + Comm. F" to support GPU halo exchange...

[alexxy/gromacs.git] / src / gromacs / domdec / gpuhaloexchange.h

diff --git a/src/gromacs/domdec/gpuhaloexchange.h b/src/gromacs/domdec/gpuhaloexchange.h
index b20ad0a1808003f32255ddfe86208312f57bdb13..9482009410c060ca3ebe880e5473fcdb89f7b76e 100644 (file)
--- a/src/gromacs/domdec/gpuhaloexchange.h
+++ b/src/gromacs/domdec/gpuhaloexchange.h
@@ -49,6 +49,7 @@
 #include "gromacs/utility/gmxmpi.h"
 
 struct gmx_domdec_t;
+struct gmx_wallcycle;
 class DeviceContext;
 class DeviceStream;
 class GpuEventSynchronizer;
@@ -86,13 +87,15 @@ public:
      * \param [in]    streamLocal              local NB CUDA stream.
      * \param [in]    streamNonLocal           non-local NB CUDA stream.
      * \param [in]    pulse                    the communication pulse for this instance
+     * \param [in]    wcycle                   The wallclock counter
      */
     GpuHaloExchange(gmx_domdec_t*        dd,
                     MPI_Comm             mpi_comm_mysim,
                     const DeviceContext& deviceContext,
                     const DeviceStream&  streamLocal,
                     const DeviceStream&  streamNonLocal,
-                    int                  pulse);
+                    int                  pulse,
+                    gmx_wallcycle*       wcycle);
     ~GpuHaloExchange();
 
     /*! \brief