diff --git a/paddle_inference/paddle/include/paddle_engine.h b/paddle_inference/paddle/include/paddle_engine.h index 1bdb62655aced39c1839d69f4bb297f293a6073e..79aaa8733bc0ab8f06c3e238d5abd65497c980d2 100644 --- a/paddle_inference/paddle/include/paddle_engine.h +++ b/paddle_inference/paddle/include/paddle_engine.h @@ -268,7 +268,7 @@ class PaddleInferenceEngine : public EngineCore { config.EnableGpuMultiStream(); } } - config.EnableTensorRtEngine(1 << 20, + config.EnableTensorRtEngine(1 << 25, max_batch, min_subgraph_size, precision_type,