From ec4ae3c2b5cb0e83fb667f14f832ea94f68ef075 Mon Sep 17 00:00:00 2001 From: Pasha Date: Wed, 26 Oct 2022 20:05:28 +0000 Subject: added .wait() for device->host transfer --- simplex-dev/src/kernel.cpp | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) (limited to 'simplex-dev') 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 &inAHost, std::vector &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 diff = end - start; -- cgit v1.2.1