Skip to content

Commit c556694

Browse files
committed
Merge remote-tracking branch 'origin/perf/ptv3-maxauxstreams1' into perf/trt-plugins-cache-sort-workspace-size
2 parents 78db54b + f4f0404 commit c556694

1 file changed

Lines changed: 7 additions & 0 deletions

File tree

perception/autoware_ptv3/lib/ptv3_trt.cpp

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -291,6 +291,13 @@ void PTv3TRT::initTrt(const tensorrt_common::TrtCommonConfig & trt_config)
291291
network_trt_ptr_ = std::make_unique<autoware::tensorrt_common::TrtCommon>(
292292
trt_config, std::make_shared<autoware::tensorrt_common::Profiler>(),
293293
std::vector<std::string>{config_.plugins_path_});
294+
295+
auto trt_builder_config = network_trt_ptr_->getBuilderConfig();
296+
if (trt_builder_config == nullptr) {
297+
throw std::runtime_error("Failed to get builder config from TRT engine." + config_.plugins_path_);
298+
}
299+
300+
trt_builder_config->setMaxAuxStreams(1);
294301

295302
if (!network_trt_ptr_->setup(std::move(profile_dims_ptr), std::move(network_io_ptr))) {
296303
throw std::runtime_error("Failed to setup TRT engine." + config_.plugins_path_);

0 commit comments

Comments
 (0)