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

[Backend] Add fixed size optimization for transformer model (#1430)

Browse Source 
Add enable_fixed_size_opt flag
This commit is contained in:
Jack Zhou
2023-02-24 09:45:04 +08:00
committed by GitHub
parent d3845eb4e1
commit 524c85745b
3 changed files with 23 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
fastdeploy/runtime/backends/paddle/option_pybind.cc
View File

@@ -36,6 +36,8 @@ void BindPaddleOption(pybind11::module& m) {
BindIpuOption(m);
pybind11::class_<PaddleBackendOption>(m, "PaddleBackendOption")
.def(pybind11::init())
.def_readwrite("enable_fixed_size_opt",
&PaddleBackendOption::enable_fixed_size_opt)
.def_readwrite("enable_log_info", &PaddleBackendOption::enable_log_info)
.def_readwrite("enable_mkldnn", &PaddleBackendOption::enable_mkldnn)
.def_readwrite("enable_trt", &PaddleBackendOption::enable_trt)