【Paddle Tensor】fix converter old ir issues -1 (#70849) (#71643)

* Update manipulation.py

* Update manipulation.py

* Update manipulation.py

* fix temporal_shift

* Update others.py

* Update others.py

* update

* fix codestyle

* Update auto_scan_test.py

* Update auto_scan_test.py

* modify softmax

* fix codestyle

* Update CMakeLists.txt

* fix codestyle

* Update manipulation.py

* Update manipulation.py

* Update manipulation.py

* update

* Update pir.cc

* Update pir.cc

* fix codestyle

* Update pir.cc

* fix codestyle

* fix

* delete old_softmax

* Update pir.cc

* fix codestyle

* fix tile

* fix tile

* fix expand

* fix codestyle

* fix range

* fix tile

* Update test_trt_convert_temporal_shift_deprecated.py

* fix tile

* fix codestyle

* add expand

Author: PolaKuma

paddle/fluid/pybind/pir.cc

@@ -984,6 +984,11 @@ void BindOperation(py::module *m) {
                  attrs_dict[pair.first.c_str()] =
                      pair.second.dyn_cast<OperationDistAttribute>();
                } else {
+                 if (pair.second.isa<pir::FloatAttribute>()) {
+                   VLOG(2) << "The value is stored with float32 precision, "
+                              "which may cause precision issues for higher "
+                              "precision requirements.";
+                 }
                  attrs_dict[pair.first.c_str()] =
                      paddle::dialect::GetAttributeData(pair.second);
                }

python/paddle/tensorrt/impls/manipulation.py

@@ -1049,11 +1049,17 @@ def tile_converter(network, paddle_op, inputs):
         repeat_rank = len(repeat_times)
     else:
         repeat_tensor = inputs[1]
-        repeat_tensor = resize_to_1d(
-            network, repeat_tensor, name=[paddle_op.name(), 'repeat_tensor']
-        )
-        repeat_shape = paddle_op.operands()[1].source().shape
-        repeat_rank = repeat_shape[0]
+        if isinstance(repeat_tensor, list):
+            repeat_rank = len(repeat_tensor)
+            repeat_tensor = trt_concat(
+                network, repeat_tensor, name=[paddle_op.name(), 'repeat_tensor']
+            )
+        else:
+            repeat_tensor = resize_to_1d(
+                network, repeat_tensor, name=[paddle_op.name(), 'repeat_tensor']
+            )
+            repeat_shape = paddle_op.operands()[1].source().shape
+            repeat_rank = repeat_shape[0]
 
     if rank > repeat_rank:
         one_rank_tensor = add_1D_constant_layer(

python/paddle/tensorrt/impls/others.py

@@ -323,7 +323,8 @@ def share_data_converter(network, paddle_op, inputs):
 @converter_registry.register("pd_op.temporal_shift", trt_version="8.x")
 def temporal_shift_converter(network, paddle_op, inputs):
     input_tensor = inputs[0]
-    shift_ratio = paddle_op.attrs()["shift_ratio"]
+    # Add a small bias to shift_ratio to mitigate floating point precision errors
+    shift_ratio = paddle_op.attrs()["shift_ratio"] + 1e-7
     T = paddle_op.attrs()["seg_num"]
     data_format = paddle_op.attrs().get("data_format", "NCHW")
 

test/ir/inference/CMakeLists.txt

@@ -169,7 +169,10 @@ if(WITH_GPU AND TENSORRT_FOUND)
   #set_tests_properties(test_trt_multiclass_nms_op PROPERTIES TIMEOUT 200)
 
   set_tests_properties(test_trt_ops_fp32_mix_precision PROPERTIES TIMEOUT 300)
+  set_tests_properties(test_trt_convert_expand_v2 PROPERTIES TIMEOUT 1000)
   set_tests_properties(test_trt_convert_unary PROPERTIES TIMEOUT 600)
+  set_tests_properties(test_trt_convert_temporal_shift_deprecated
+                       PROPERTIES TIMEOUT 1000)
   set_tests_properties(test_trt_convert_pool2d PROPERTIES TIMEOUT 600)
   set_tests_properties(test_trt_convert_slice PROPERTIES TIMEOUT 600)