add arguments for trt config

paddle/fluid/inference/analysis/analyzer_tester.cc

@@ -37,6 +37,14 @@ TEST(Analyzer, analysis_without_tensorrt) {
 TEST(Analyzer, analysis_with_tensorrt) {
   FLAGS_IA_enable_tensorrt_subgraph_engine = true;
   Argument argument;
+  int* minimum_subgraph_size = new int(0);
+  int* max_batch_size = new int(3);
+  int* workspace_size = new int(1 << 20);
+  std::string* precision_mode = new std::string("FP32");
+  argument.Set<int>("minimum_subgraph_size", minimum_subgraph_size);
+  argument.Set<int>("max_batch_size", max_batch_size);
+  argument.Set<int>("workspace_size", workspace_size);
+  argument.Set<std::string>("precision_mode", precision_mode);
   argument.fluid_model_dir.reset(new std::string(FLAGS_inference_model_dir));
   Analyzer analyser;
   analyser.Run(&argument);

paddle/fluid/inference/analysis/subgraph_splitter.cc

@@ -309,7 +309,7 @@ void SubGraphFuse::operator()() { ReplaceNodesWithSubGraphs(); }
 void SubGraphFuse::ReplaceNodesWithSubGraphs() {
   auto subgraphs = SubGraphSplitter(graph_, node_inside_subgraph_teller_)();
   for (auto &subgraph : subgraphs) {
-    if (subgraph.size() <= argument_->Get<int>("minimun_subgraph_size"))
+    if (subgraph.size() <= argument_->Get<int>("minimum_subgraph_size"))
       continue;
     std::unordered_set<Node *> subgraph_uniq(subgraph.begin(), subgraph.end());
     // replace this sub-graph with the first node. Two steps: 1. Create a Block

paddle/fluid/inference/analysis/subgraph_splitter_tester.cc

@@ -68,7 +68,7 @@ TEST(SubGraphSplitter, Fuse) {
   auto dfg = ProgramDescToDFG(desc);
   Argument argument;
   int* minmum_subgraph_size = new int(3);
-  argument.Set<int>("minmum_subgraph_size", minmum_subgraph_size);
+  argument.Set<int>("minimum_subgraph_size", minmum_subgraph_size);
 
   size_t count0 = dfg.nodes.size();
 

paddle/fluid/inference/analysis/tensorrt_subgraph_pass_tester.cc

@@ -36,6 +36,14 @@ TEST(TensorRTSubGraphPass, main) {
   };
 
   Argument argument(FLAGS_inference_model_dir);
+  int* minimum_subgraph_size = new int(0);
+  int* max_batch_size = new int(3);
+  int* workspace_size = new int(1 << 20);
+  std::string* precision_mode = new std::string("FP32");
+  argument.Set<int>("minimun_subgraph_size", minimum_subgraph_size);
+  argument.Set<int>("max_batch_size", max_batch_size);
+  argument.Set<int>("workspace_size", workspace_size);
+  argument.Set<std::string>("precision_mode", precision_mode);
 
   DFG_GraphvizDrawPass::Config config{FLAGS_dot_dir, "origin"};
   DFG_GraphvizDrawPass::Config config1{FLAGS_dot_dir, "fusion"};

paddle/fluid/inference/api/api_tensorrt_subgraph_engine.cc

@@ -94,7 +94,7 @@ class TensorRTSubgraphPredictor : public NativePaddlePredictor {
     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);
-    argument.Set<int>("minimun_subgraph_size", minimum_subgraph_size);
+    argument.Set<int>("minimum_subgraph_size", minimum_subgraph_size);
     argument.Set<int>("max_batch_size", max_batch_size);
     argument.Set<int>("workspace_size", workspace_size);
     argument.Set<std::string>("precision_mode", precision_mode);