Ensure all work is complete on device before freeing memory. (#724)

libocca · Nov 30, 2023 · 5285d7b · 5285d7b
1 parent b0c3227
commit 5285d7b
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/occa/internal/modes/dpcpp/buffer.cpp b/src/occa/internal/modes/dpcpp/buffer.cpp
@@ -15,11 +15,13 @@ namespace occa {
     buffer::~buffer() {
       if (!isWrapped && ptr) {
         auto& dpcpp_device = getDpcppDevice(modeDevice);
+        // Unlike CUDA and HIP, freeing memory in SYCL is not
+        // synchronous across all streams.
+        dpcpp_device.finishAll();
         OCCA_DPCPP_ERROR("Memory: Freeing SYCL alloc'd memory",
                          ::sycl::free(ptr,dpcpp_device.dpcppContext));
       }
       ptr = nullptr;
-      size = 0;
     }
 
     void buffer::malloc(udim_t bytes) {