Position buffer ops in CUDA

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

diff --git a/src/gromacs/ewald/pme_gpu_internal.cpp b/src/gromacs/ewald/pme_gpu_internal.cpp
index 27da475ac8c3e0eec0e428d4afbaee39252d87ab..a8560a7a020485d3b77de284615f2ca15a774436 100644 (file)
--- a/src/gromacs/ewald/pme_gpu_internal.cpp
+++ b/src/gromacs/ewald/pme_gpu_internal.cpp
@@ -236,6 +236,10 @@ void pme_gpu_copy_input_coordinates(const PmeGpu *pmeGpu, const rvec *h_coordina
     copyToDeviceBuffer(&pmeGpu->kernelParams->atoms.d_coordinates, h_coordinatesFloat,
                        0, pmeGpu->kernelParams->atoms.nAtoms * DIM,
                        pmeGpu->archSpecific->pmeStream, pmeGpu->settings.transferKind, nullptr);
+    // FIXME: sync required since the copied data will be used by PP stream when using single GPU for both
+    //        Remove after adding the required event-based sync between the above H2D and the transform kernel
+    pme_gpu_synchronize(pmeGpu);
+
 #endif
 }
 
@@ -1267,3 +1271,16 @@ void pme_gpu_gather(PmeGpu                *pmeGpu,
 
     pme_gpu_copy_output_forces(pmeGpu);
 }
+
+void * pme_gpu_get_kernelparam_coordinates(const PmeGpu *pmeGpu)
+{
+    if (pmeGpu && pmeGpu->kernelParams)
+    {
+        return pmeGpu->kernelParams->atoms.d_coordinates;
+    }
+    else
+    {
+        return nullptr;
+    }
+
+}