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

add option to upload model function, pipeline creation honors option use flags, setting allocator per extractor do not make much sense

Browse Source
This commit is contained in:
nihuini
2019-06-05 19:03:39 +08:00
parent 83d7154be8
commit e09607bc22
49 changed files with 281 additions and 275 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/command.cpp
+4 -3
View File
@@ -17,6 +17,7 @@
#if NCNN_VULKAN
#include <stdio.h>
#include "option.h"
namespace ncnn {
@@ -841,16 +842,16 @@ VkTransfer::~VkTransfer()
{
}
void VkTransfer::record_upload(const Mat& src, VkMat& dst)
void VkTransfer::record_upload(const Mat& src, VkMat& dst, const Option& opt)
{
if (src.elemsize / src.packing == 4)
{
if (vkdev->info.support_fp16_storage || (vkdev->info.support_fp16_packed && src.packing % 4 == 0))
if (opt.use_fp16_storage || (opt.use_fp16_packed && src.packing % 4 == 0))
{
Mat src_fp16;
cast_float32_to_float16(src, src_fp16);
record_upload(src_fp16, dst);
record_upload(src_fp16, dst, opt);
return;
}