Position buffer ops in CUDA

[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 efe4a60235e557cac1b0c84ffaddf09bbd74f509..07d15cdf1a55130b6eb9e53ec6992874503ab443 100644 (file)
--- a/src/gromacs/ewald/pme_gpu.cpp
+++ b/src/gromacs/ewald/pme_gpu.cpp
@@ -391,3 +391,12 @@ void pme_gpu_reinit_computation(const gmx_pme_t *pme,
     wallcycle_sub_stop(wcycle, ewcsLAUNCH_GPU_PME);
     wallcycle_stop(wcycle, ewcLAUNCH_GPU);
 }
+
+void *pme_gpu_get_device_x(const gmx_pme_t *pme)
+{
+    if (!pme || !pme_gpu_active(pme))
+    {
+        return nullptr;
+    }
+    return pme_gpu_get_kernelparam_coordinates(pme->gpu);
+}