fix: Adjust reflection pad test cases to prevent runtime errors (#3088)

Author: chohk88

py/torch_tensorrt/dynamo/conversion/impl/pad.py

@@ -79,7 +79,7 @@ def get_padded_shape_tensors(
                     ).get_output(0)
                 )
         padded_shape_tensor = impl.cat.cat(
-            ctx, target, source_ir, f"{name}_cat", slices, 0
+            ctx, target, source_ir, f"{name}_cat_dim_{i}", slices, 0
         )
 
     start_indices_tensor = get_trt_tensor(

tests/py/dynamo/conversion/test_pad_aten.py

@@ -146,9 +146,9 @@ def forward(self, input):
         [
             (
                 "4d",
-                (1, 1, 1, 1),
                 (2, 2, 2, 2),
-                (3, 3, 3, 3),
+                (4, 4, 4, 4),
+                (6, 6, 6, 6),
                 torch.float,
                 (1, 1, 2, 2),
             ),
@@ -199,9 +199,9 @@ def forward(self, input):
         [
             (
                 "5d",
-                (1, 1, 1, 1, 1),
                 (2, 2, 2, 2, 2),
-                (3, 3, 3, 3, 3),
+                (4, 4, 4, 4, 4),
+                (6, 6, 6, 6, 6),
                 torch.float,
                 (1, 2, 2, 1, 1, 2),
             ),