GithubStar/ncnn
1
0
Fork 0
mirror of https://github.com/Tencent/ncnn.git synced 2026-10-24 09:45:20 +08:00
Code Issues Packages Projects Releases Wiki Activity

add pytest option (#2587)

Browse Source
This commit is contained in:
Cai Shanli
2021-01-11 23:26:44 +08:00
committed by GitHub
parent e243edf991
commit e246dba39b
2 changed files with 147 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/src/main.cpp
+3 -2
View File
@@ -183,20 +183,21 @@ PYBIND11_MODULE(ncnn, m)
.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)
.def_readwrite("use_bf16_storage", &Option::use_bf16_storage)
.def_readwrite("use_fp16_packed", &Option::use_fp16_packed)
.def_readwrite("use_fp16_storage", &Option::use_fp16_storage)
.def_readwrite("use_fp16_arithmetic", &Option::use_fp16_arithmetic)
.def_readwrite("use_int8_packed", &Option::use_int8_packed)
.def_readwrite("use_int8_storage", &Option::use_int8_storage)
.def_readwrite("use_int8_arithmetic", &Option::use_int8_arithmetic)
.def_readwrite("use_packing_layout", &Option::use_packing_layout)
.def_readwrite("use_shader_pack8", &Option::use_shader_pack8)
.def_readwrite("use_image_storage", &Option::use_image_storage)
.def_readwrite("use_subgroup_basic", &Option::use_subgroup_basic)
.def_readwrite("use_subgroup_vote", &Option::use_subgroup_vote)
.def_readwrite("use_subgroup_ballot", &Option::use_subgroup_ballot)
.def_readwrite("use_subgroup_shuffle", &Option::use_subgroup_shuffle)
.def_readwrite("use_image_storage", &Option::use_image_storage)
.def_readwrite("use_bf16_storage", &Option::use_bf16_storage)
.def_readwrite("use_tensor_storage", &Option::use_tensor_storage)
.def_readwrite("use_weight_fp16_storage", &Option::use_weight_fp16_storage);
py::class_<Mat> mat(m, "Mat", py::buffer_protocol());