diff --git a/paddle/fluid/inference/tests/api/CMakeLists.txt b/paddle/fluid/inference/tests/api/CMakeLists.txt index 56b222c75ceec4351b7777fc6f67ab8cde497081..52e711444d199c6c74536d8e503018d3565ec00e 100644 --- a/paddle/fluid/inference/tests/api/CMakeLists.txt +++ b/paddle/fluid/inference/tests/api/CMakeLists.txt @@ -621,13 +621,13 @@ if(WITH_GPU) endif() if(WITH_GPU AND TENSORRT_FOUND) - set_tests_properties(trt_resnext_test PROPERTIES TIMEOUT 120) - set_tests_properties(trt_quant_int8_yolov3_r50_test PROPERTIES TIMEOUT 120) + set_tests_properties(trt_resnext_test PROPERTIES TIMEOUT 150) + set_tests_properties(trt_quant_int8_yolov3_r50_test PROPERTIES TIMEOUT 150) set_tests_properties(trt_resnet50_test PROPERTIES TIMEOUT 120) - set_tests_properties(trt_cascade_rcnn_test PROPERTIES TIMEOUT 120) - set_tests_properties(test_trt_dynamic_shape_ernie_ser_deser PROPERTIES TIMEOUT 120) - set_tests_properties(test_trt_dynamic_shape_ernie_fp16_ser_deser PROPERTIES TIMEOUT 120) - set_tests_properties(test_trt_dynamic_shape_ernie PROPERTIES TIMEOUT 120) + set_tests_properties(trt_cascade_rcnn_test PROPERTIES TIMEOUT 150) + set_tests_properties(test_trt_dynamic_shape_ernie_ser_deser PROPERTIES TIMEOUT 150) + set_tests_properties(test_trt_dynamic_shape_ernie_fp16_ser_deser PROPERTIES TIMEOUT 150) + set_tests_properties(test_trt_dynamic_shape_ernie PROPERTIES TIMEOUT 150) endif() if(WITH_MKLDNN) diff --git a/paddle/fluid/inference/tests/api/trt_dynamic_shape_ernie_test.cc b/paddle/fluid/inference/tests/api/trt_dynamic_shape_ernie_test.cc index 6bf34484e5dffddc189fedaff20dbd6557dfcf70..875175d9e83d78cccdf089714eae3742246658c3 100644 --- a/paddle/fluid/inference/tests/api/trt_dynamic_shape_ernie_test.cc +++ b/paddle/fluid/inference/tests/api/trt_dynamic_shape_ernie_test.cc @@ -139,7 +139,7 @@ TEST(AnalysisPredictor, no_fp16) { TEST(AnalysisPredictor, fp16) { #ifdef TRT_PLUGIN_FP16_AVALIABLE std::vector result = {0.598, 0.219, 0.182}; - trt_ernie(true, result, 3e-3); + trt_ernie(true, result, 4e-3); #endif }