a copy of PaddlePaddle#51412

Author: jiweibo

test/cpp/inference/api/CMakeLists.txt

@@ -1027,6 +1027,14 @@ if(WITH_TESTING AND WITH_INFERENCE_API_TEST)
     else()
       target_link_libraries(test_analyzer_capi_exp_xpu paddle_inference_c)
     endif()
+    inference_analysis_test(
+      trt_rebind_stream_test
+      SRCS
+      trt_rebind_stream_test.cc
+      EXTRA_DEPS
+      paddle_inference_shared
+      ARGS
+      --infer_model=${TRT_MODEL_INSTALL_DIR}/trt_inference_test_models)
 
     set(TRT_MODEL_QUANT_RESNET_DIR
         "${INFERENCE_DEMO_INSTALL_DIR}/small_quant_model")
@@ -1378,6 +1386,8 @@ if(WITH_TESTING AND WITH_INFERENCE_API_TEST)
   endif()
   if(WITH_GPU AND TENSORRT_FOUND)
     set_tests_properties(trt_mobilenet_test PROPERTIES TIMEOUT 120)
+    set_tests_properties(trt_rebind_stream_test PROPERTIES TIMEOUT 120 LABELS
+                                                           "RUN_TYPE=EXCLUSIVE")
     if(WITH_MKLDNN)
       set_tests_properties(test_analyzer_bfloat16_resnet50 PROPERTIES TIMEOUT
                                                                       120)

paddle/fluid/inference/tests/api/trt_rebind_stream_test.cc renamed to test/cpp/inference/api/trt_rebind_stream_test.cc

@@ -16,7 +16,7 @@ limitations under the License. */
 #include <gtest/gtest.h>
 
 #include "gflags/gflags.h"
-#include "paddle/fluid/inference/tests/api/trt_test_helper.h"
+#include "test/cpp/inference/api/tester_helper.h"
 
 namespace paddle {
 namespace inference {