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

Add trt max workspace setting (#308)

Browse Source 
* add trt max workspace setting

* fix set trt max workspace
This commit is contained in:
Jack Zhou
2022-09-30 09:54:34 +08:00
committed by GitHub
parent 4a4c37aa97
commit dd365fb721
5 changed files with 13 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
fastdeploy/pybind/runtime.cc
View File

@@ -36,6 +36,7 @@ void BindRuntime(pybind11::module& m) {
&RuntimeOption::SetPaddleMKLDNNCacheSize)
.def("set_lite_power_mode", &RuntimeOption::SetLitePowerMode)
.def("set_trt_input_shape", &RuntimeOption::SetTrtInputShape)
.def("set_trt_max_workspace_size", &RuntimeOption::SetTrtMaxWorkspaceSize)
.def("enable_trt_fp16", &RuntimeOption::EnableTrtFP16)
.def("disable_trt_fp16", &RuntimeOption::DisableTrtFP16)
.def("set_trt_cache_file", &RuntimeOption::SetTrtCacheFile)