diff --git a/python/paddle/fluid/tests/unittests/ir/inference/auto_scan_test.py b/python/paddle/fluid/tests/unittests/ir/inference/auto_scan_test.py index 8a68dc7a0ac7d654eedf72a472d90762fd7b2dd4..bb8c6e73fdefada768b2297470db1c9b4ffa32fc 100755 --- a/python/paddle/fluid/tests/unittests/ir/inference/auto_scan_test.py +++ b/python/paddle/fluid/tests/unittests/ir/inference/auto_scan_test.py @@ -697,8 +697,7 @@ class TrtLayerAutoScanTest(AutoScanTest): pred_config_deserialize, feed_data) except Exception as e: self.fail_log( - str(prog_config) + ' vs ' + self.inference_config_str( - pred_config) + + self.inference_config_str(pred_config) + '\033[1;31m \nERROR INFO: {}\033[0m'.format(str(e))) if not ignore_flag: status = False diff --git a/python/paddle/fluid/tests/unittests/ir/inference/test_trt_convert_reduce_mean.py b/python/paddle/fluid/tests/unittests/ir/inference/test_trt_convert_reduce_mean.py index b09ae80555e08da9976cd0f19bdf1962ba388c7f..ba648042dabf755326e709a46218caa9d857b506 100644 --- a/python/paddle/fluid/tests/unittests/ir/inference/test_trt_convert_reduce_mean.py +++ b/python/paddle/fluid/tests/unittests/ir/inference/test_trt_convert_reduce_mean.py @@ -120,7 +120,7 @@ class TrtConvertReduceMeanTest(TrtLayerAutoScanTest): attrs, False), 1e-5 self.trt_param.precision = paddle_infer.PrecisionType.Half yield self.create_inference_config(), generate_trt_nodes_num( - attrs, False), (1e-4, 1e-4) + attrs, False), (5e-4, 5e-4) # for dynamic_shape generate_dynamic_shape(attrs) @@ -129,7 +129,7 @@ class TrtConvertReduceMeanTest(TrtLayerAutoScanTest): True), 1e-5 self.trt_param.precision = paddle_infer.PrecisionType.Half yield self.create_inference_config(), generate_trt_nodes_num( - attrs, True), (1e-4, 1e-4) + attrs, True), (5e-4, 5e-4) pass