Added barrier

2021-03-09 10:00:33 +00:00 · 2021-03-09 10:00:33 +00:00 · c9bd406c8b
commit c9bd406c8b
parent 0ea3a79123
1 changed files with 6 additions and 0 deletions
--- a/src/OpTensorSyncLocal.cpp
+++ b/src/OpTensorSyncLocal.cpp
@ -38,6 +38,12 @@ OpTensorSyncLocal::record(const vk::CommandBuffer& commandBuffer)
                                        vk::PipelineStageFlagBits::eTransfer);

            this->mTensors[i]->recordCopyFromDeviceToStaging(commandBuffer);
+
+            this->mTensors[i]->recordPrimaryBufferMemoryBarrier(commandBuffer,
+                                        vk::AccessFlagBits::eTransferWrite,
+                                        vk::AccessFlagBits::eHostRead,
+                                        vk::PipelineStageFlagBits::eTransfer,
+                                        vk::PipelineStageFlagBits::eHost);
        }
    }
 }