add wait and throw

uxlfoundation · Dec 10, 2024 · 4d66692 · 4d66692
1 parent 8ab34b6
commit 4d66692
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/cpp/oneapi/dal/backend/memory.hpp b/cpp/oneapi/dal/backend/memory.hpp
@@ -222,6 +222,7 @@ inline sycl::event memcpy_host2usm(sycl::queue& queue,
     ONEDAL_ASSERT(is_known_usm(queue, dest_usm));
 
     sycl::event memcpy_event = memcpy(queue, dest_usm, src_host, size, deps);
+    memcpy_event.wait_and_throw();
     return memcpy_event;
 }
 
@@ -233,6 +234,7 @@ inline sycl::event memcpy_usm2host(sycl::queue& queue,
     ONEDAL_ASSERT(is_known_usm(queue, src_usm));
 
     sycl::event memcpy_event = memcpy(queue, dest_host, src_usm, size, deps);
+    memcpy_event.wait_and_throw();
     return memcpy_event;
 }