refine uttest of api_tensorrt_subgraph_engine

478eeabd · nhzlx · 791aa7f4 · 478eeabd
显示空白变更内容
内联并排

Showing with 1 addition and 0 deletion

paddle/fluid/inference/api/api_tensorrt_subgraph_engine_tester.cc ...luid/inference/api/api_tensorrt_subgraph_engine_tester.cc +1 -0

未找到文件。
--- a/paddle/fluid/inference/api/api_tensorrt_subgraph_engine_tester.cc
+++ b/paddle/fluid/inference/api/api_tensorrt_subgraph_engine_tester.cc
@@ -37,6 +37,7 @@ void CompareTensorRTWithFluid(bool enable_tensorrt) {
  config1.use_gpu = true;
  config1.fraction_of_gpu_memory = 0.3;
  config1.device = 0;
+  config1.max_batch_size = 10;

  auto predictor0 =
      CreatePaddlePredictor<NativeConfig, PaddleEngineKind::kNative>(config0);