From 478eeabdd488fb155242629f39516ac3897d0bce Mon Sep 17 00:00:00 2001
From: nhzlx <zlx_hg@163.com>
Date: Mon, 27 Aug 2018 05:11:41 +0000
Subject: [PATCH] refine uttest of api_tensorrt_subgraph_engine

---
 .../fluid/inference/api/api_tensorrt_subgraph_engine_tester.cc   | 1 +
 1 file changed, 1 insertion(+)

diff --git a/paddle/fluid/inference/api/api_tensorrt_subgraph_engine_tester.cc b/paddle/fluid/inference/api/api_tensorrt_subgraph_engine_tester.cc
index 8f1a72316..9e7425edd 100644
--- 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);
-- 
GitLab