Make use of the DeviceStreamManager

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

diff --git a/src/gromacs/nbnxm/gpu_data_mgmt.h b/src/gromacs/nbnxm/gpu_data_mgmt.h
index 8e114d1c65cb9affd4c5c9ee77f60dfd21c03168..21fc8174b56d494617b6e47ba5251563db467b05 100644 (file)
--- a/src/gromacs/nbnxm/gpu_data_mgmt.h
+++ b/src/gromacs/nbnxm/gpu_data_mgmt.h
@@ -50,9 +50,6 @@
 #include "gromacs/gpu_utils/gpu_macros.h"
 #include "gromacs/mdtypes/locality.h"
 
-class DeviceContext;
-class DeviceStream;
-
 struct NbnxmGpu;
 struct gmx_gpu_info_t;
 struct DeviceInformation;
@@ -62,12 +59,19 @@ struct NbnxnPairlistGpu;
 struct PairlistParams;
 struct interaction_const_t;
 
+class DeviceStream;
+
+namespace gmx
+{
+class DeviceStreamManager;
+}
+
 namespace Nbnxm
 {
 
 /** Initializes the data structures related to GPU nonbonded calculations. */
 GPU_FUNC_QUALIFIER
-NbnxmGpu* gpu_init(const DeviceContext gmx_unused& deviceContext,
+NbnxmGpu* gpu_init(const gmx::DeviceStreamManager gmx_unused& deviceStreamManager,
                    const interaction_const_t gmx_unused* ic,
                    const PairlistParams gmx_unused& listParams,
                    const nbnxn_atomdata_t gmx_unused* nbat,