Correctly sync all streams in device::finishAll(). (#723)

libocca · Jan 24, 2024 · 789e9fc · 789e9fc
1 parent 06e32fd
commit 789e9fc
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 3 deletions.
diff --git a/src/occa/internal/core/device.cpp b/src/occa/internal/core/device.cpp
@@ -85,8 +85,14 @@ namespace occa {
   }
 
   void modeDevice_t::finishAll() const {
-    for(auto* stream : streams) {
-      if(stream) stream->finish();
+    if (streamRing.head) {
+      auto* start = static_cast<modeStream_t*>(streamRing.head);
+      auto* current = start;
+      // Every device 
+      do {
+        current->finish();
+        current = static_cast<modeStream_t*>(current->rightRingEntry);
+      } while (current != start);
     }
   }
 

diff --git a/src/occa/internal/core/device.hpp b/src/occa/internal/core/device.hpp
@@ -21,7 +21,6 @@ namespace occa {
     gc::ring_t<modeStreamTag_t> streamTagRing;
 
     stream currentStream;
-    std::vector<modeStream_t*> streams;
 
     udim_t bytesAllocated;
     udim_t maxBytesAllocated;