add wait

alpaka-group · Jul 23, 2024 · e80097e · e80097e
1 parent c60ef7a
commit e80097e
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/example/matrixAddWithMdspan/src/matrixAddMdSpan.cpp b/example/matrixAddWithMdspan/src/matrixAddMdSpan.cpp
@@ -127,6 +127,7 @@ auto example(TAccTag const&) -> int
     // Copy data to device, use directly host buffers (not mdspans used to fill the data)
     alpaka::memcpy(queue, bufDevA, bufHostA);
     alpaka::memcpy(queue, bufDevB, bufHostB);
+    alpaka::wait(queue);
 
     // Create mdspan views for device buffers using alpaka::experimental::getMdSpan
     auto mdDevA = alpaka::experimental::getMdSpan(bufDevA);
@@ -146,7 +147,6 @@ auto example(TAccTag const&) -> int
 
     // Copy result back to host
     alpaka::memcpy(queue, bufHostC, bufDevC);
-    // This wait is not necessary if the queue is a blocking queue
     alpaka::wait(queue);
 
     // Verify the result

diff --git a/example/matrixMulWithMdspan/src/matrixMulMdSpan.cpp b/example/matrixMulWithMdspan/src/matrixMulMdSpan.cpp
@@ -138,6 +138,7 @@ auto example(TAccTag const&) -> int
     // Copy data to device, use directly host buffers (not mdspans used to fill the data)
     alpaka::memcpy(queue, bufDevA, bufHostA);
     alpaka::memcpy(queue, bufDevB, bufHostB);
+    alpaka::wait(queue);
 
     // Create mdspan views for device buffers using alpaka::experimental::getMdSpan
     auto mdDevA = alpaka::experimental::getMdSpan(bufDevA);
@@ -157,7 +158,6 @@ auto example(TAccTag const&) -> int
 
     // Copy result back to host
     alpaka::memcpy(queue, bufHostC, bufDevC);
-    // This wait is not necessary if the queue is a blocking queue
     alpaka::wait(queue);
 
     // Verify the result