Decouple coordinates buffer management from buffer ops in NBNXM

[alexxy/gromacs.git] / src / gromacs / ewald / pme_gpu.cpp

diff --git a/src/gromacs/ewald/pme_gpu.cpp b/src/gromacs/ewald/pme_gpu.cpp
index 1f43a9fe5f2893a248232549c8321962e6eced20..2bf148a59320ff92d11fc2ac49fa69531707efcc 100644 (file)
--- a/src/gromacs/ewald/pme_gpu.cpp
+++ b/src/gromacs/ewald/pme_gpu.cpp
@@ -420,12 +420,9 @@ void pme_gpu_reinit_computation(const gmx_pme_t *pme,
     wallcycle_stop(wcycle, ewcLAUNCH_GPU);
 }
 
-void *pme_gpu_get_device_x(const gmx_pme_t *pme)
+DeviceBuffer<float> pme_gpu_get_device_x(const gmx_pme_t *pme)
 {
-    if (!pme || !pme_gpu_active(pme))
-    {
-        return nullptr;
-    }
+    GMX_ASSERT((pme && pme_gpu_active(pme)), "PME GPU coordinates buffer was requested from uninitialized PME module");
     return pme_gpu_get_kernelparam_coordinates(pme->gpu);
 }