apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2025-11-01 20:32:52 +08:00
Code Issues Actions 6 Packages Projects Releases Wiki Activity

Update option.h

Browse Source
This commit is contained in:
Jason
2023-02-22 11:28:04 +08:00
committed by GitHub
parent cc9031c17e
commit dd527388bc
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
fastdeploy/runtime/backends/paddle/option.h
View File

@@ -67,10 +67,12 @@ struct PaddleBackendOption {
/// initialize memory size(MB) for GPU /// initialize memory size(MB) for GPU
int gpu_mem_init_size = 100; int gpu_mem_init_size = 100;
/// Disable type of operators run on TensorRT
void DisableTrtOps(const std::vector<std::string>& ops) { void DisableTrtOps(const std::vector<std::string>& ops) {
trt_disabled_ops_.insert(trt_disabled_ops_.end(), ops.begin(), ops.end()); trt_disabled_ops_.insert(trt_disabled_ops_.end(), ops.begin(), ops.end());
} }
/// Delete pass by name
void DeletePass(const std::string& pass_name) { void DeletePass(const std::string& pass_name) {
delete_pass_names.push_back(pass_name); delete_pass_names.push_back(pass_name);
} }