[Inference]Support control trt_glog_info in PIR (#72429)

* add_debug_trt

* Update paddle_analysis_config.h

* fix codestyle

* fix codestyle

* Update tensorrt_test_base.py

* fix

* Update paddle_analysis_config.h

* Update paddle_analysis_config.h

* update

* fix codestyle

* update

* fix codestyle

Author: PolaKuma

paddle/fluid/inference/api/analysis_config.cc

@@ -19,7 +19,6 @@
 
 #include "glog/logging.h"
 #include "paddle/common/errors.h"
-#include "paddle/common/flags.h"
 #include "paddle/fluid/inference/api/helper.h"
 #include "paddle/fluid/inference/api/paddle_analysis_config.h"
 #include "paddle/fluid/inference/api/paddle_pass_builder.h"

python/paddle/tensorrt/converter.py

@@ -135,7 +135,8 @@ def convert_subgraph_to_trt(self, program, group_op):
         if self.trt_config is not None and self.trt_config.ops_run_float:
             _logger.info(f"force_fp32_ops: {trt_manager.get_force_fp32_ops()}")
 
-        _logger.info(f"start process {group_op}")
+        if not self.trt_config.disable_loggling:
+            _logger.info(f"start process {group_op}")
 
         operations = next(iter(group_op.blocks())).ops
         input_values, output_values = self.find_graph_inputs_outputs(group_op)
@@ -326,9 +327,10 @@ def convert_subgraph_to_trt(self, program, group_op):
                 # constant/parameter condition, needn't get min/opt/max shape
                 continue
             input_name = trt_input.name
-            _logger.info(
-                f"set shape of {value}, op is: {value.get_defining_op()}"
-            )
+            if not self.trt_config.disable_loggling:
+                _logger.info(
+                    f"set shape of {value}, op is: {value.get_defining_op()}"
+                )
             min_shape = []
             opt_shape = []
             max_shape = []
@@ -372,22 +374,24 @@ def convert_subgraph_to_trt(self, program, group_op):
                         value, True, paddle.base.core.ShapeMode.kMAX
                     )
             if not trt_input.is_shape_tensor:
-                _logger.info(f"set min_shape of {value} as {min_shape}")
-                _logger.info(f"set opt_shape of {value} as {opt_shape}")
-                _logger.info(f"set max_shape of {value} as {max_shape}")
+                if not self.trt_config.disable_loggling:
+                    _logger.info(f"set min_shape of {value} as {min_shape}")
+                    _logger.info(f"set opt_shape of {value} as {opt_shape}")
+                    _logger.info(f"set max_shape of {value} as {max_shape}")
                 profile.set_shape(
                     input_name, min=min_shape, opt=opt_shape, max=max_shape
                 )
             else:
-                _logger.info(
-                    f"set min_value of shape input: {value} as {min_value}"
-                )
-                _logger.info(
-                    f"set opt_value of shape input: {value} as {opt_value}"
-                )
-                _logger.info(
-                    f"set max_value of shape input: {value} as {max_value}"
-                )
+                if not self.trt_config.disable_loggling:
+                    _logger.info(
+                        f"set min_value of shape input: {value} as {min_value}"
+                    )
+                    _logger.info(
+                        f"set opt_value of shape input: {value} as {opt_value}"
+                    )
+                    _logger.info(
+                        f"set max_value of shape input: {value} as {max_value}"
+                    )
                 profile.set_shape_input(
                     input_name, min=min_value, opt=opt_value, max=max_value
                 )
@@ -610,7 +614,8 @@ def convert(self, network, paddle_op, inputs):
     def convert_program_to_trt(self):
         for op in self.program.global_block().ops:
             if op.name() == "cinn_op.group" or op.name() == "builtin.group":
-                _logger.info(f"start process {op.name()}")
+                if not self.trt_config.disable_loggling:
+                    _logger.info(f"start process {op.name()}")
                 self.engine_num += 1
                 new_out = self.convert_subgraph_to_trt(self.program, op)
                 orin_out_values = op.results()

python/paddle/tensorrt/export.py

@@ -242,6 +242,7 @@ def __init__(
         workspace_size: int | None = 1 << 30,
         use_cuda_graph: bool | None = False,
         refit_params_path: str | None = None,
+        disable_loggling: bool | None = True,
     ) -> None:
         """
         A class for configuring TensorRT optimizations.
@@ -273,6 +274,8 @@ def __init__(
                 Specify whether TensorRT enables cuda_graph during the optimization process (default is false).
             refit_params_path(str, optional):
                 The path to the weights that need to be refitted.
+            disable_loggling (bool, optional):
+                Specifies whether to enable GLOG info output during the optimization process (default is true).
         Returns:
             None
 
@@ -330,6 +333,7 @@ def __init__(
         self.workspace_size = workspace_size
         self.use_cuda_graph = use_cuda_graph
         self.refit_params_path = refit_params_path
+        self.disable_loggling = disable_loggling
         if self.refit_params_path:
             self.disable_passes.append("constant_folding_pass")
         paddle.framework.set_flags(

test/tensorrt/tensorrt_test_base.py

@@ -303,6 +303,7 @@ def check_trt_result(self, rtol=1e-5, atol=1e-5, precision_mode="fp32"):
                 max_input_shape=self.max_shape,
             )
             trt_config = TensorRTConfig(inputs=[input])
+            trt_config.disable_loggling = False
             if precision_mode == "fp16":
                 trt_config.precision_mode = PrecisionMode.FP16