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

Merge branch 'develop' into set_stream_infer-shareExData

Browse Source
This commit is contained in:
Wang Bojun
2023-02-14 14:23:33 +08:00
committed by GitHub
parent 727fdc5863 3aafceed07
commit a0fc0d5e49
14 changed files with 385 additions and 253 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
fastdeploy/runtime/option_pybind.cc
View File

@@ -35,6 +35,7 @@ void BindOption(pybind11::module& m) {
.def(pybind11::init())
.def("set_model_path", &RuntimeOption::SetModelPath)
.def("set_model_buffer", &RuntimeOption::SetModelBuffer)
.def("set_encryption_key", &RuntimeOption::SetEncryptionKey)
.def("use_gpu", &RuntimeOption::UseGpu)
.def("use_cpu", &RuntimeOption::UseCpu)
.def("use_rknpu2", &RuntimeOption::UseRKNPU2)