GithubStar/ncnn
1
0
Fork 0
mirror of https://github.com/Tencent/ncnn.git synced 2026-10-28 11:47:38 +08:00
Code Issues Packages Projects Releases Wiki Activity

add missing barrier for transfer dst, fix softmax pack4, fix #932

Browse Source
This commit is contained in:
nihuini
2019-04-24 16:16:26 +08:00
parent 4729ea3505
commit 5d86014d9c
6 changed files with 43 additions and 58 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/command.cpp
+2
View File
@@ -224,6 +224,7 @@ void VkCompute::record_download(const VkMat& m)
void VkCompute::record_clone(const VkMat& src, const VkMat& dst)
{
record_prepare_transfer_barrier(src);
record_prepare_transfer_barrier(dst);
if (vkdev->info.support_VK_KHR_push_descriptor)
return copy_buffer(src.buffer(), src.buffer_offset(), dst.buffer(), dst.buffer_offset(), src.total() * src.elemsize);
@@ -249,6 +250,7 @@ void VkCompute::record_copy_region(const VkMat& src, const VkMat& dst, const VkB
void VkCompute::record_copy_regions(const VkMat& src, const VkMat& dst, const std::vector<VkBufferCopy>& regions)
{
record_prepare_transfer_barrier(src);
record_prepare_transfer_barrier(dst);
if (vkdev->info.support_VK_KHR_push_descriptor)
return copy_buffer_regions(src.buffer(), dst.buffer(), regions);