fix fp16 io (#54129)

* fix fp16 io

* disable precision test

paddle/fluid/framework/ir/auto_mixed_precision_pass.cc

@@ -509,6 +509,7 @@ void AutoMixedPrecisionPass::UpdateOpPrecision() const {
         // when op_1 only support cpu kernel. if op_2's intput var is op_1's
         // output var, then op_2 should not run at low precision.
         if (GetOpOriginalType(op_type) != "feed" &&
+            GetOpOriginalType(op_type) != "tensorrt_engine" &&
             !KernelSupportPrecision(
                 GetOpOriginalType(op_type), backend_, phi::DataType::FLOAT32)) {
           for (auto* out_var_node : op_node->outputs) {

test/ir/inference/test_trt_inference_fp16_io.py

@@ -66,10 +66,11 @@ class TestEnableLowPrecisionIO:
             fp32_output = self.get_fp32_output()
             fp16_output = self.get_fp16_output()
 
-            # np.testing.assert_allclose(
-            #     fp32_output.numpy().flatten(),
-            #     fp16_output.numpy().flatten(),
-            # )
+        # if os.name == 'posix':
+        #     np.testing.assert_allclose(
+        #         fp32_output.numpy().flatten(),
+        #         fp16_output.numpy().flatten(),
+        #     )
 
 
 class TestEnableLowPrecisionIOWithGPU(
@@ -105,6 +106,7 @@ class TestEnableLowPrecisionIOWithTRTAllGraph(
             use_static=False,
             use_calib_mode=False,
         )
+        config.enable_tuned_tensorrt_dynamic_shape()
         config.enable_memory_optim()
         config.enable_low_precision_io(low_precision_io)
         config.disable_glog_info()
@@ -129,6 +131,7 @@ class TestEnableLowPrecisionIOWithTRTSubGraph(
             use_static=False,
             use_calib_mode=False,
         )
+        config.enable_tuned_tensorrt_dynamic_shape()
         config.enable_memory_optim()
         config.enable_low_precision_io(low_precision_io)
         config.exp_disable_tensorrt_ops(["flatten_contiguous_range"])