diff --git a/fastdeploy/pybind/main.cc.in b/fastdeploy/pybind/main.cc.in index 5e11ee808..5fde0082a 100755 --- a/fastdeploy/pybind/main.cc.in +++ b/fastdeploy/pybind/main.cc.in @@ -181,9 +181,6 @@ PYBIND11_MODULE(@PY_LIBRARY_NAME@, m) { m.def_submodule("text", "Text module of FastDeploy."); BindText(text_module); #endif - auto rknpu2_module = - m.def_submodule("rknpu2", "RKNPU2 config module of FastDeploy."); - BindRKNPU2Config(rknpu2_module); } } // namespace fastdeploy diff --git a/fastdeploy/pybind/rknpu2_config_pybind.cc b/fastdeploy/runtime/backends/rknpu2/rknpu2_config_pybind.cc similarity index 97% rename from fastdeploy/pybind/rknpu2_config_pybind.cc rename to fastdeploy/runtime/backends/rknpu2/rknpu2_config_pybind.cc index d7f95de8a..028b7a9f1 100644 --- a/fastdeploy/pybind/rknpu2_config_pybind.cc +++ b/fastdeploy/runtime/backends/rknpu2/rknpu2_config_pybind.cc @@ -14,7 +14,7 @@ #include "fastdeploy/pybind/main.h" #include "fastdeploy/runtime/backends/rknpu2/option.h" namespace fastdeploy { -void BindRKNPU2Config(pybind11::module& m) { +void BindRKNPU2Option(pybind11::module& m) { pybind11::enum_( m, "CpuName", pybind11::arithmetic(), "CpuName for inference.") .value("RK356X", fastdeploy::rknpu2::CpuName::RK356X) diff --git a/fastdeploy/runtime/option_pybind.cc b/fastdeploy/runtime/option_pybind.cc index e5443d894..b5ac9e5b9 100644 --- a/fastdeploy/runtime/option_pybind.cc +++ b/fastdeploy/runtime/option_pybind.cc @@ -22,7 +22,7 @@ void BindOrtOption(pybind11::module& m); void BindTrtOption(pybind11::module& m); void BindPaddleOption(pybind11::module& m); void BindPorosOption(pybind11::module& m); - +void BindRKNPU2Option(pybind11::module& m); void BindOption(pybind11::module& m) { BindLiteOption(m); BindOpenVINOOption(m); @@ -30,6 +30,7 @@ void BindOption(pybind11::module& m) { BindTrtOption(m); BindPaddleOption(m); BindPorosOption(m); + BindRKNPU2Option(m); pybind11::class_(m, "RuntimeOption") .def(pybind11::init())