igl | vulkan | Fix bug in staging device

Reviewed By: corporateshark, rameshviswanathan Differential Revision: D66967542 fbshipit-source-id: dd801255a39772d2165f9f264cf5dcfdf5d16da4
facebook · Dec 10, 2024 · 434bac5 · 434bac5
1 parent c1f4fd5
commit 434bac5
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/igl/vulkan/VulkanStagingDevice.cpp b/src/igl/vulkan/VulkanStagingDevice.cpp
@@ -174,7 +174,8 @@ VulkanStagingDevice::MemoryRegion VulkanStagingDevice::nextFreeBlock(VkDeviceSiz
   auto regionItr = regions_.begin();
   while (regionItr != regions_.end()) {
     // if requested size is available or if contiguous memory is not requested
-    if (regionItr->size >= requestedAlignedSize || (!contiguous)) {
+    if ((regionItr->size >= requestedAlignedSize || !contiguous) &&
+        (immediate_->isReady(regionItr->handle))) {
       allocatedSize = std::min(regionItr->size, requestedAlignedSize);
       break;
     }