GithubStar/ncnn
1
0
Fork 0
mirror of https://github.com/Tencent/ncnn.git synced 2026-10-25 18:30:16 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix memorydata test on nvidia gpu

Browse Source
This commit is contained in:
nihui
2020-06-03 00:31:58 +08:00
parent f87f21779f
commit 478e15d7dc
3 changed files with 6 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/command.cpp
+2 -2
View File
@@ -2507,7 +2507,7 @@ VkTransfer::~VkTransfer()
}
}
void VkTransfer::record_upload(const Mat& src, VkMat& dst, const Option& opt)
void VkTransfer::record_upload(const Mat& src, VkMat& dst, const Option& opt, bool flatten)
{
// NCNN_LOGE("record_upload src = %d | %d %d %d @ %d", src.dims, src.w, src.h, src.c, src.elempack);
@@ -2525,7 +2525,7 @@ void VkTransfer::record_upload(const Mat& src, VkMat& dst, const Option& opt)
}
}
Mat src_flattened = src.reshape(src.w * src.h * src.c);
Mat src_flattened = flatten ? src.reshape(src.w * src.h * src.c) : src;
// create dst
dst.create_like(src_flattened, opt.blob_vkallocator);