From 1f6c9dbad41c9a4a2a5fa66af372c6f9074403b5 Mon Sep 17 00:00:00 2001 From: nhzlx Date: Fri, 21 Sep 2018 08:06:11 +0000 Subject: [PATCH] fix typo --- paddle/fluid/inference/api/api_tensorrt_subgraph_engine.cc | 2 +- paddle/fluid/inference/api/paddle_inference_api.h | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/paddle/fluid/inference/api/api_tensorrt_subgraph_engine.cc b/paddle/fluid/inference/api/api_tensorrt_subgraph_engine.cc index 8be444949..ce147eb5d 100644 --- a/paddle/fluid/inference/api/api_tensorrt_subgraph_engine.cc +++ b/paddle/fluid/inference/api/api_tensorrt_subgraph_engine.cc @@ -90,7 +90,7 @@ class TensorRTSubgraphPredictor : public NativePaddlePredictor { // Analyze inference_program Argument argument; - int* minimum_subgraph_size = new int(config_.minimun_subgraph_size); + int* minimum_subgraph_size = new int(config_.minimum_subgraph_size); int* max_batch_size = new int(config_.max_batch_size); int* workspace_size = new int(config_.workspace_size); std::string* precision_mode = new std::string(config_.precision_mode); diff --git a/paddle/fluid/inference/api/paddle_inference_api.h b/paddle/fluid/inference/api/paddle_inference_api.h index 084da823e..bacb319e2 100644 --- a/paddle/fluid/inference/api/paddle_inference_api.h +++ b/paddle/fluid/inference/api/paddle_inference_api.h @@ -155,7 +155,7 @@ struct TensorRTConfig : public NativeConfig { // We set this variable to control the minimum number of nodes in the // subgraph, 3 as // default value. - int minimun_subgraph_size = 3; + int minimum_subgraph_size = 3; // Reserved configuration // We just support "FP32" now, "FP16" and "INT8" will be supported. std::string precision_mode = "FP32"; -- GitLab