apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2025-10-07 01:22:59 +08:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

update pybind

Browse Source
This commit is contained in:
Zheng-Bicheng
2023-03-07 15:50:49 +08:00
parent 190d254a95
commit 458523fc29
3 changed files with 3 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
fastdeploy/pybind/main.cc.in
View File

@@ -181,9 +181,6 @@ PYBIND11_MODULE(@PY_LIBRARY_NAME@, m) {
m.def_submodule("text", "Text module of FastDeploy."); m.def_submodule("text", "Text module of FastDeploy.");
BindText(text_module); BindText(text_module);
#endif #endif
auto rknpu2_module =
m.def_submodule("rknpu2", "RKNPU2 config module of FastDeploy.");
BindRKNPU2Config(rknpu2_module);
} }
} // namespace fastdeploy } // namespace fastdeploy

2
fastdeploy/pybind/rknpu2_config_pybind.cc → fastdeploy/runtime/backends/rknpu2/rknpu2_config_pybind.cc
View File

@@ -14,7 +14,7 @@
#include "fastdeploy/pybind/main.h" #include "fastdeploy/pybind/main.h"
#include "fastdeploy/runtime/backends/rknpu2/option.h" #include "fastdeploy/runtime/backends/rknpu2/option.h"
namespace fastdeploy { namespace fastdeploy {
void BindRKNPU2Config(pybind11::module& m) { void BindRKNPU2Option(pybind11::module& m) {
pybind11::enum_<fastdeploy::rknpu2::CpuName>( pybind11::enum_<fastdeploy::rknpu2::CpuName>(
m, "CpuName", pybind11::arithmetic(), "CpuName for inference.") m, "CpuName", pybind11::arithmetic(), "CpuName for inference.")
.value("RK356X", fastdeploy::rknpu2::CpuName::RK356X) .value("RK356X", fastdeploy::rknpu2::CpuName::RK356X)

3
fastdeploy/runtime/option_pybind.cc
View File

@@ -22,7 +22,7 @@ void BindOrtOption(pybind11::module& m);
void BindTrtOption(pybind11::module& m); void BindTrtOption(pybind11::module& m);
void BindPaddleOption(pybind11::module& m); void BindPaddleOption(pybind11::module& m);
void BindPorosOption(pybind11::module& m); void BindPorosOption(pybind11::module& m);
void BindRKNPU2Option(pybind11::module& m);
void BindOption(pybind11::module& m) { void BindOption(pybind11::module& m) {
BindLiteOption(m); BindLiteOption(m);
BindOpenVINOOption(m); BindOpenVINOOption(m);
@@ -30,6 +30,7 @@ void BindOption(pybind11::module& m) {
BindTrtOption(m); BindTrtOption(m);
BindPaddleOption(m); BindPaddleOption(m);
BindPorosOption(m); BindPorosOption(m);
BindRKNPU2Option(m);
pybind11::class_<RuntimeOption>(m, "RuntimeOption") pybind11::class_<RuntimeOption>(m, "RuntimeOption")
.def(pybind11::init()) .def(pybind11::init())