fix [RUNTIME][VULKAN] vkBuffer released before memory copy command se…

…nd to GPU (apache#5388) (apache#5418)
dpankratz · Apr 24, 2020 · 5b16db6 · 5b16db6
1 parent d8e09af
commit 5b16db6
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/runtime/vulkan/vulkan.cc b/src/runtime/vulkan/vulkan.cc
@@ -189,6 +189,10 @@ class VulkanDeviceAPI final : public DeviceAPI {
   }
 
   void FreeDataSpace(TVMContext ctx, void* ptr) final {
+    // Before releasing the vkBuffer, call sync to
+    // finish all the vulkan commands that reference the buffer.
+    StreamSync(ctx, nullptr);
+
     const auto& vctx = context(ctx.device_id);
     auto* pbuf = static_cast<VulkanBuffer*>(ptr);
     vkDestroyBuffer(vctx.device, pbuf->buffer, nullptr);