Update tensorrt_test_base.py

PolaKuma · PolaKuma · commit 2f8d3ae6beeb · 2025-04-24T10:55:51.000+08:00
diff --git a/test/tensorrt/tensorrt_test_base.py b/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.glog_info = True
             if precision_mode == "fp16":
                 trt_config.precision_mode = PrecisionMode.FP16
 

Original file line number	Diff line number	Diff line change
`@@ -303,6 +303,7 @@ def check_trt_result(self, rtol=1e-5, atol=1e-5, precision_mode="fp32"):`
`303`	`303`	`max_input_shape=self.max_shape,`
`304`	`304`	`)`
`305`	`305`	`trt_config = TensorRTConfig(inputs=[input])`
	`306`	`+ trt_config.glog_info = True`
`306`	`307`	`if precision_mode == "fp16":`
`307`	`308`	`trt_config.precision_mode = PrecisionMode.FP16`
`308`	`309`