Fix direct calls to SDAG entry methods
[namd.git] / src / CudaPmeSolver.ci
index 440148c..d9056b4 100644 (file)
@@ -30,7 +30,7 @@ module CudaPmeSolver {
                        }
                };
                entry void recvDeviceBuffer(DeviceDataMsg *msg);
-               entry void recvDeviceBuffers() {
+               entry void recvDeviceBuffers(const CkCallback &cb) {
                        overlap {
                        for (imsg=0;imsg < numDeviceBuffers;++imsg) {
                                        when recvDeviceBuffer(DeviceDataMsg *msg) serial {
@@ -46,6 +46,7 @@ module CudaPmeSolver {
                                }
                        }
                        serial { setDeviceBuffers(); }
+                       serial { cb.send(); }
                };
        };
 
@@ -62,7 +63,7 @@ module CudaPmeSolver {
                        }
                };
                entry void recvDeviceBuffer(DeviceDataMsg *msg);
-               entry void recvDeviceBuffers() {
+               entry void recvDeviceBuffers(const CkCallback &cb) {
                        overlap {
                        for (imsgY=0;imsgY < numDeviceBuffers;++imsgY) {
                                        when recvDeviceBuffer(DeviceDataMsg *msg) serial {
@@ -78,6 +79,7 @@ module CudaPmeSolver {
                                }
                        }
                        serial { setDeviceBuffers(); }
+                       serial { cb.send(); }
                };
        };
 
@@ -101,7 +103,7 @@ module CudaPmeSolver {
                };
                entry void recvDeviceBufferZ(DeviceDataMsg *msg);
                entry void recvDeviceBufferX(DeviceDataMsg *msg);
-               entry void recvDeviceBuffers() {
+               entry void recvDeviceBuffers(const CkCallback &cb) {
                        overlap {
                        for (imsgZ=0;imsgZ < numDeviceBuffersZ;++imsgZ) {
                                        when recvDeviceBufferZ(DeviceDataMsg *msg) serial {
@@ -129,6 +131,7 @@ module CudaPmeSolver {
                                }
                        }
                        serial { setDeviceBuffers(); }
+                       serial { cb.send(); }
                };
        };
 
@@ -154,7 +157,7 @@ module CudaPmeSolver {
                        }                       
                };
                entry void recvDeviceBuffer(DeviceDataMsg *msg);
-               entry void recvDeviceBuffers() {
+               entry void recvDeviceBuffers(const CkCallback &cb) {
                        overlap {
                        for (imsgY=0;imsgY < numDeviceBuffers;++imsgY) {
                                        when recvDeviceBuffer(DeviceDataMsg *msg) serial {
@@ -175,10 +178,11 @@ module CudaPmeSolver {
                                }
                        }
                        serial { setDeviceBuffers(); }
+                       serial { cb.send(); }
                };
 
 
        };
 #endif // NAMD_CUDA
 
-};
\ No newline at end of file
+};