File tree Expand file tree Collapse file tree
perception/autoware_ptv3/lib Expand file tree Collapse file tree Original file line number Diff line number Diff 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_ );
You can’t perform that action at this time.
0 commit comments