[PIR] translate old ir to pir_program and trt program (PaddlePaddle#70364)

* support pir_trt

* fix

* fix

* fix

* fix

* fix

* fix

* fix

* fix

* fix

* fix

* fix

Author: risemeup1

paddle/fluid/pir/utils/name_analysis.cc

@@ -34,6 +34,15 @@ pir::Value GetOutputValueByName(const pir::Program &program,
         }
         value = op.operand_source(0);
       }
+    } else if (op.isa<paddle::dialect::FeedOp>() ||
+               op.isa<paddle::dialect::FetchOp>()) {
+      if (op.attribute("name") == name_attr) {
+        if (value) {
+          PADDLE_THROW(common::errors::PreconditionNotMet(
+              "More than one feed/fetch named with %s found.", name));
+        }
+        value = op.result(0);
+      }
     }
   }
   return value;

python/paddle/base/executor.py

@@ -1238,6 +1238,17 @@ def cinn_process(program):
                     op.result(0).persistable,
                 )
                 data_op_infos.append(tup)
+            if op.name() == 'pd_op.feed':
+                feed_target_name = op.attrs()["name"]
+                var_type = paddle_type_to_proto_type[op.results()[0].dtype]
+                var_shape = op.results()[0].shape
+                tup = (
+                    feed_target_name,
+                    var_type,
+                    var_shape,
+                    op.result(0).persistable,
+                )
+                data_op_infos.append(tup)
 
         return program, new_exe, data_op_infos
 

python/paddle/tensorrt/converter.py

@@ -19,10 +19,6 @@
 import numpy as np
 import tensorrt as trt
 
-# init tensorrt plugin
-trt_plugin_lib = ctypes.CDLL('libnvinfer_plugin.so')
-trt_plugin_lib.initLibNvInferPlugins(None, "")
-
 import paddle
 from paddle import pir
 from paddle.base.core import clear_shape_info, get_value_shape_range_info
@@ -83,6 +79,9 @@ def __init__(self, paddle_program, scope, trt_config=None):
         self.input_info = {}
         self.trt_output_value_map = {}
         self.engine_num = 0
+        # init tensorrt plugin
+        trt_plugin_lib = ctypes.CDLL('libnvinfer_plugin.so')
+        trt_plugin_lib.initLibNvInferPlugins(None, "")
 
     def find_graph_inputs_outputs(self, group_op):
         operations = next(iter(group_op.blocks())).ops

python/paddle/tensorrt/export.py

@@ -123,13 +123,13 @@ def generate_input_data(self):
             low, high = self.input_range
             self.input_min_data = np.random.randint(
                 low, high, size=self.min_input_shape
-            )
+            ).astype(self.input_data_type)
             self.input_optim_data = np.random.randint(
                 low, high, size=self.optim_input_shape
-            )
+            ).astype(self.input_data_type)
             self.input_max_data = np.random.randint(
                 low, high, size=self.max_input_shape
-            )
+            ).astype(self.input_data_type)
         else:
             low, high = self.input_range if self.input_range else (0, 1)
             self.input_min_data = np.random.uniform(