nihui/ncnn - ncnn - 开源协同云脑生态支撑系统

History

nihui 10b8ac68cc [WIP] vulkan compute (#618 ) * vulkan infrastructure * vkallocator and vkmat * layer interface for vulkan compute * wip... * default vulkan device, command wrapper, upload model weight in load_model to simplify layer interface * simplify command api, vkmat holds staging buffer, relu works * initialize specialization constant, simplify command dispatch, fix staging buffer copy with different shape, convolution works * init extension functions * dynamic local size and group count * group count=1 is invalid * regard device max workgroup size limit * fix relu oooops * decouple command record and staging allocation * create result blob * add pooling shader * buffer is faster than image :) * fix pooling shader * add innerproduct shader * readonly writeonly decoration * simplify buffer creation * decouple command and layer, VK_KHR_descriptor_update_template extension makes descriptor binding update easy :D * fix vulkan building issues in visual studio (#1) * fix building issues on visual studio * ignore benchmark * cancel changes * ... ... * decouple paramdict and vulkandevice * fix staging buffer destroy in model loading * remove vkdev member in option * add padding shader * simplify vulkan layer creation, simplify convolution and pooling shader for no padding, less debug output * add convolutiondepthwise and softmax shader * specialization float type, add leakyrelu * add dropout shader * add batchnorm shader * split vulkan forward * add scale shader * push constant type can be int or float * set_optimal_local_size_xyz * add eltwise shader * concat vulkan forward * fix convolution without bias * add dummy shader for concat and split, more fix ... * optional VK_KHR_descriptor_update_template and VK_KHR_push_descriptor * check VK_KHR_push_descriptor for vkCmdPushDescriptorSetWithTemplateKHR * binaryop and unaryop shader * hide raw command buffer * simple vkbenchncnn benchmark * create device with transfer queue * rename command to vkcompute, add vktransfer and layer upload_model interface * external VkMat, copy and map wrt buffer offset * command copy respect offset and size * decouple weight upload and load, simplify upload weight api, use one big staging buffer for uploading weights * fix build on android * binding count can not vary :( * barrier check state, fix sub-op destruction * declare local_size_xyz constant, fix crash on radv * fix local_size_xyz, second try * more barrier and state fix * fix softmax * reconstruct buffer memory allocator, reuse blob buffer, less verbose output * find unified memory type index * weight staging buffer allocator and weight buffer allocator, respect descriptor buffer offset alignment * use VK_KHR_descriptor_update_template for faster descriptor update if available, multithread pipeline creation * find more useful vulkan extensions and enable them * fix msvc build * respect VK_KHR_dedicated_allocation for weight buffer allocation * fix android build * fix bias name conflicts with metal * decouple pipeline and layer, building shader sources into shader module, dedicated create_pipeline api, simplify pipeline recording * drop dummy shader, inplace softmax, multiple shader module works * fix unique queue family index error * flatten support vulkan * mnasnet run * find shader module by name, each entry point per shader module, fix attribute/id conflict on moltenvk * some minor changes * add some high level api * use dedicated transfer queue to upload weight model * prefer mappable buffer on unified memory * global pooling and convolution fc, reuse staging buffer * implement ring-buffer style blob allocator, add VkBufferMemory capacity * use blob allocator for workspace blob, it works fine :) * vulkan option off * Update layer.cpp * fix build with vulkan off * less verbose output, fix crash on vulkan_compute off * merge benchncnn tool * allocator clear api, use new weight buffer allocator per net * add default locked allocator * mapped mat ptr api, persistent mapped memory works generally :) * travis ci linux vulkan * travis ci vulkan wip ... * more gpu wip ... * more gpu wip ... * wip... * wip... * wip... ... * wip... ios vulkan build... * find glslangValidator on ios build * use dynamic moltenvk library * travis ci wip ... * ios simulator does not support metal at all * fix cpu only extractor * optimize workgroup size, first try * optimize workgroup size, second try * conv1x1s1d1 vec4 * revert build system * fix ncnn2mem build * fix ncnn2mem build		7 years ago
..
caffe	recent protobuf library requires c++11 features, try enabling c++11 for caffe2ncnn onnx2ncnn tensorflow2ncnn	7 years ago

darknet	add info about darknet2ncnn	7 years ago

mxnet	convert mxnet Pad	7 years ago

onnx	convert onnx Flatten	7 years ago

plugin	Request add ncnn::Mat support for Visual Studio Image Watch Tools (#625)	7 years ago

pytorch	add pytorch tool readme	8 years ago

tensorflow	recent protobuf library requires c++11 features, try enabling c++11 for caffe2ncnn onnx2ncnn tensorflow2ncnn	7 years ago

CMakeLists.txt	[WIP] vulkan compute (#618)	7 years ago

ncnn2mem.cpp	layer holds bottom blob scale, depthwise convolution read group scales	7 years ago

No Description

C++ C Python Text Protocol Buffer other

nihuini@tencent.com shuizhuyuanluo@126.com 248857878@qq.com youhanmir@gmail.com leeward.xie@gmail.com 379489343@qq.com lzx1413@live.cn abdel.younes@nviso.ch ericliu612@gmail.com 314377460@qq.com liushouda@qq.com diegogomes_770@hotmail.com myhouseng@gmail.com tiancai.ye@outlook.com zuyu@cs.wisc.edu liuchang5@tom.com penjilu@gmail.com tedder59@users.noreply.github.com 617878975@qq.com shunli0521.hi@163.com cqhuawei_rd@163.com 791136190@qq.com 820169199@qq.com xsl@impower-tech.com dev.chandan.rai@gmail.com

Contributors (25+) All

Contributors (25+)
All