GithubStar/ncnn
1
0
Fork 0
mirror of https://github.com/Tencent/ncnn.git synced 2026-10-27 03:13:52 +08:00
Code Issues Packages Projects Releases Wiki Activity

Python ncnn with vulkan (#5020)

Browse Source 
* add option for enabling vulkan in setup.py
* load winograd type options for pybind.
* enable the usage of humbletim/install-vulkan-sdk in ci
* vulkan-loader with release build
* python setup.py install --vulkan=on
* moltenVK for x86_64 macos
* remove universal2  temporarily
This commit is contained in:
Zhenyu ZHAO
2023-10-23 10:50:15 +08:00
committed by GitHub
parent 80b3b9c6f0
commit a6decad15d
4 changed files with 374 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/src/main.cpp
+3
View File
@@ -185,6 +185,9 @@ PYBIND11_MODULE(ncnn, m)
#endif // NCNN_VULKAN
.def_readwrite("openmp_blocktime", &Option::openmp_blocktime)
.def_readwrite("use_winograd_convolution", &Option::use_winograd_convolution)
.def_readwrite("use_winograd23_convolution", &Option::use_winograd23_convolution)
.def_readwrite("use_winograd43_convolution", &Option::use_winograd43_convolution)
.def_readwrite("use_winograd63_convolution", &Option::use_winograd63_convolution)
.def_readwrite("use_sgemm_convolution", &Option::use_sgemm_convolution)
.def_readwrite("use_int8_inference", &Option::use_int8_inference)
.def_readwrite("use_vulkan_compute", &Option::use_vulkan_compute)