From c9bd406c8b2fb268459da5e8716bb59c25b0f95b Mon Sep 17 00:00:00 2001 From: Alejandro Saucedo Date: Tue, 9 Mar 2021 10:00:33 +0000 Subject: [PATCH] Added barrier --- src/OpTensorSyncLocal.cpp | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/src/OpTensorSyncLocal.cpp b/src/OpTensorSyncLocal.cpp index 8a99c780b..5e653154a 100644 --- 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); } } }