Take over management of OpenCL context from PME and NBNXM

[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 9eac3f7c78977948d58a545e589c5f413dbd5b7c..822852786c80e0417c320b84b6b76d41aab46a0c 100644 (file)
--- a/src/gromacs/nbnxm/gpu_data_mgmt.h
+++ b/src/gromacs/nbnxm/gpu_data_mgmt.h
@@ -50,6 +50,8 @@
 #include "gromacs/gpu_utils/gpu_macros.h"
 #include "gromacs/mdtypes/locality.h"
 
+class DeviceContext;
+
 struct NbnxmGpu;
 struct gmx_gpu_info_t;
 struct DeviceInformation;
@@ -65,6 +67,7 @@ namespace Nbnxm
 /** Initializes the data structures related to GPU nonbonded calculations. */
 GPU_FUNC_QUALIFIER
 NbnxmGpu* gpu_init(const DeviceInformation gmx_unused* deviceInfo,
+                   const DeviceContext gmx_unused& deviceContext,
                    const interaction_const_t gmx_unused* ic,
                    const PairlistParams gmx_unused& listParams,
                    const nbnxn_atomdata_t gmx_unused* nbat,