Skip to content

Commit 409c5dc

Browse files
[Backend]Add default value for TRT SetShape() (PaddlePaddle#1712)
* add GPL lisence * add GPL-3.0 lisence * add GPL-3.0 lisence * add GPL-3.0 lisence * support yolov8 * add pybind for yolov8 * add yolov8 readme * add cpp benchmark * add cpu and gpu mem * public part split * add runtime mode * fixed bugs * add cpu_thread_nums * deal with comments * deal with comments * deal with comments * rm useless code * add FASTDEPLOY_DECL * add FASTDEPLOY_DECL * fixed for windows * mv rss to pss * mv rss to pss * Update utils.cc * use thread to collect mem * Add ResourceUsageMonitor * rm useless code * fixed bug * fixed typo * update ResourceUsageMonitor * fixed bug * fixed bug * add note for ResourceUsageMonitor * deal with comments * add macros * deal with comments * deal with comments * deal with comments * re-lint * rm pmap and use mem api * rm pmap and use mem api * add mem api * Add PrintBenchmarkInfo func * Add PrintBenchmarkInfo func * Add PrintBenchmarkInfo func * deal with comments * fixed enable_paddle_to_trt * add log for paddle_trt * support ppcls benchmark * use new trt option api * update benchmark info * simplify benchmark.cc * simplify benchmark.cc * deal with comments * Add ppseg && ppocr benchmark * add OCR rec img * add ocr benchmark * fixed trt shape * add trt shape * resolve conflict * add ENABLE_BENCHMARK define * Add ClassifyDiff * Add Resize for ClassifyResult * deal with comments * add convert info script * resolve conflict * Add SaveBenchmarkResult func * fixed bug * fixed bug * fixed bug * add config.txt for option * fixed bug * fixed bug * fixed bug * add benchmark.sh * mv thread_nums from 8 to 1 * deal with comments * deal with comments * fixed readme * deal with comments * add all platform shell * Update config.arm.txt * Update config.gpu.txt * Update config.x86.txt * fixed printinfo bug * rm proxy * add more model support * all backend config.txt * deal with comments * Add MattingDiff compare * fixed predict bug * adjust warmup/repeat times * add e2e/mem configs * fixed typo * open collect_mem * fixed typo * add trt cache option * fixed bug * fixed repeat times * test for benchmark * test for det benchmark * for benchmark * fixed for x86 * add h2d and d2h config * renmae txt file * add dynamic shape for pp_trt * fixed typo * Update option.h * add collect shape * add default value for SetShape() --------- Co-authored-by: DefTruth <[email protected]>
1 parent d43e901 commit 409c5dc

File tree

1 file changed

+2
-2
lines changed
  • fastdeploy/runtime/backends/tensorrt

1 file changed

+2
-2
lines changed

fastdeploy/runtime/backends/tensorrt/option.h

+2-2
Original file line numberDiff line numberDiff line change
@@ -46,8 +46,8 @@ struct TrtBackendOption {
4646
*/
4747
void SetShape(const std::string& tensor_name,
4848
const std::vector<int32_t>& min,
49-
const std::vector<int32_t>& opt,
50-
const std::vector<int32_t>& max) {
49+
const std::vector<int32_t>& opt = std::vector<int32_t>(),
50+
const std::vector<int32_t>& max = std::vector<int32_t>()) {
5151
min_shape[tensor_name].clear();
5252
max_shape[tensor_name].clear();
5353
opt_shape[tensor_name].clear();

0 commit comments

Comments
 (0)