apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2025-10-05 16:48:03 +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

3
python/fastdeploy/runtime.py
View File

@@ -125,6 +125,9 @@ class RuntimeOption:
def disable_trt_fp16(self):
return self._option.disable_trt_fp16()
def set_trt_max_workspace_size(self, trt_max_workspace_size):
return self._option.set_trt_max_workspace_size(trt_max_workspace_size)
def __repr__(self):
attrs = dir(self._option)
message = "RuntimeOption(\n"