Make DeviceStream into a class

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

diff --git a/src/gromacs/domdec/gpuhaloexchange.h b/src/gromacs/domdec/gpuhaloexchange.h
index 851e3d19833031fd2e669039ae1731eaa8dd4365..b20ad0a1808003f32255ddfe86208312f57bdb13 100644 (file)
--- a/src/gromacs/domdec/gpuhaloexchange.h
+++ b/src/gromacs/domdec/gpuhaloexchange.h
@@ -50,6 +50,7 @@
 
 struct gmx_domdec_t;
 class DeviceContext;
+class DeviceStream;
 class GpuEventSynchronizer;
 
 namespace gmx
@@ -89,8 +90,8 @@ public:
     GpuHaloExchange(gmx_domdec_t*        dd,
                     MPI_Comm             mpi_comm_mysim,
                     const DeviceContext& deviceContext,
-                    void*                streamLocal,
-                    void*                streamNonLocal,
+                    const DeviceStream&  streamLocal,
+                    const DeviceStream&  streamNonLocal,
                     int                  pulse);
     ~GpuHaloExchange();