Implement alternating GPU wait

[alexxy/gromacs.git] / src / gromacs / ewald / pme.cu

diff --git a/src/gromacs/ewald/pme.cu b/src/gromacs/ewald/pme.cu
index aec0074d2dfc9185c92a557bcf557b4231bd59f1..f677c7de37440cbc90e8a4c29a68e2ac9ededfb7 100644 (file)
--- a/src/gromacs/ewald/pme.cu
+++ b/src/gromacs/ewald/pme.cu
@@ -373,6 +373,11 @@ void pme_gpu_free_fract_shifts(const PmeGpu *pmeGpu)
                             pmeGpu->deviceInfo);
 }
 
+bool pme_gpu_stream_query(const PmeGpu *pmeGpu)
+{
+    return haveStreamTasksCompleted(pmeGpu->archSpecific->pmeStream);
+}
+
 void pme_gpu_copy_input_gather_grid(const PmeGpu *pmeGpu, float *h_grid)
 {
     const size_t gridSize = pmeGpu->archSpecific->realGridSize * sizeof(float);