Skip to content

Commit 6858363

Browse files
NHZlXPaddle CI
authored andcommitted
refine uttest of api_tensorrt_subgraph_engine
(cherry-pick from commit 478eeab)
1 parent ec56c75 commit 6858363

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

paddle/fluid/inference/api/api_tensorrt_subgraph_engine_tester.cc

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -37,6 +37,7 @@ void CompareTensorRTWithFluid(bool enable_tensorrt) {
3737
config1.use_gpu = true;
3838
config1.fraction_of_gpu_memory = 0.3;
3939
config1.device = 0;
40+
config1.max_batch_size = 10;
4041

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

0 commit comments

Comments
 (0)