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

drop x86 avx2 fp16 (#3568)

Browse Source
This commit is contained in:
nihui
2022-02-16 22:45:57 +08:00
committed by GitHub
parent 6b2495cc24
commit 920aa79f04
20 changed files with 33 additions and 3191 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/src/main.cpp
+1 -3
View File
@@ -197,8 +197,7 @@ PYBIND11_MODULE(ncnn, m)
.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_tensor_storage", &Option::use_tensor_storage)
.def_readwrite("use_weight_fp16_storage", &Option::use_weight_fp16_storage);
.def_readwrite("use_tensor_storage", &Option::use_tensor_storage);
py::class_<Mat> mat(m, "Mat", py::buffer_protocol());
mat.def(py::init<>())
@@ -954,7 +953,6 @@ PYBIND11_MODULE(ncnn, m)
.def_readwrite("support_bf16_storage", &Layer::support_bf16_storage)
.def_readwrite("support_fp16_storage", &Layer::support_fp16_storage)
.def_readwrite("support_image_storage", &Layer::support_image_storage)
.def_readwrite("support_weight_fp16_storage", &Layer::support_weight_fp16_storage)
.def("forward", (int (Layer::*)(const std::vector<Mat>&, std::vector<Mat>&, const Option&) const) & Layer::forward,
py::arg("bottom_blobs"), py::arg("top_blobs"), py::arg("opt"))
.def("forward", (int (Layer::*)(const Mat&, Mat&, const Option&) const) & Layer::forward,