added .wait() for device->host transfer

1 files changed, 4 insertions, 4 deletions

diff --git a/simplex-dev/src/kernel.cpp b/simplex-dev/src/kernel.cpp
index 86bd98e..4ef0a9b 100644
--- a/simplex-dev/src/kernel.cpp
+++ b/simplex-dev/src/kernel.cpp
@@ -199,10 +199,10 @@ double RunKernel(queue &q, std::vector<T> &inAHost, std::vector<T> &inBHost,
 		});
 	}).wait();
 
-	q.memcpy(inAHost.data(), inADevice, inAHost.size()*sizeof(T));
-	q.memcpy(inBHost.data(), inBDevice, inBHost.size()*sizeof(T));
-	q.memcpy(inCHost.data(), inCDevice, inCHost.size()*sizeof(T));
-	q.memcpy(resultFlags.data(), inResultFlagsDevice, resultFlags.size()*sizeof(int));
+	q.memcpy(inAHost.data(), inADevice, inAHost.size()*sizeof(T)).wait();
+	q.memcpy(inBHost.data(), inBDevice, inBHost.size()*sizeof(T)).wait();
+	q.memcpy(inCHost.data(), inCDevice, inCHost.size()*sizeof(T)).wait();
+	q.memcpy(resultFlags.data(), inResultFlagsDevice, resultFlags.size()*sizeof(int)).wait();
 
 	auto end = std::chrono::high_resolution_clock::now();
 	std::chrono::duration<double, std::milli> diff = end - start;