Add tests & update xfail_sets

Signed-off-by: Zahid Wakeel <zahid.wakeel@multicorewareinc.com>

Author: zahidwx

projects/pt1/e2e_testing/xfail_sets.py

@@ -840,6 +840,8 @@
     "ReflectionPad3dModuleRight_basic",
     "ReflectionPad3dModuleFront_basic",
     "ReflectionPad3dModuleBack_basic",
+    "ReplicationPad1dModule_2DInput_basic",
+    "ReplicationPad1dModule_3DInput_basic",
     "ReplicationPad2dModule_basic",
     "ReplicationPad2dModule_bottom0",
     "ReplicationPad2dModule_left0",
@@ -3896,6 +3898,8 @@
     "ScaledDotProductAttentionSameDynamicModule_basic",
     "ScaledDotProductAttentionSameModule_basic",
     "ScaledDotProductAttentionGQAModule_basic",
+    "ReplicationPad1dModule_2DInput_basic",
+    "ReplicationPad1dModule_3DInput_basic",
 }
 
 ONNX_TOSA_CRASHING_SET = {
@@ -4725,6 +4729,8 @@
     "ReshapeCollapseModule_basic",
     "ReshapeDynamicModule_basic",
     "ReshapeExpandModule_basic",
+    "ReplicationPad1dModule_2DInput_basic",
+    "ReplicationPad1dModule_3DInput_basic",
     "RollModule_basic",
     "RsubIntModule_noalpha_basic",
     "ScalarConstantTupleModule_basic",

projects/pt1/python/torch_mlir_e2e_test/test_suite/basic.py

@@ -685,26 +685,6 @@ def ReplicationPad2dModule_left0(module, tu: TestUtils):
 # ==============================================================================
 
 
-class ReplicationPad1dModule(torch.nn.Module):
-    def __init__(self):
-        super().__init__()
-
-    @export
-    @annotate_args(
-        [
-            None,
-            ([-1, -1, -1], torch.float32, True),
-        ]
-    )
-    def forward(self, x):
-        return torch.ops.aten.replication_pad1d(x, [3, 5])
-
-
-@register_test_case(module_factory=lambda: ReplicationPad1dModule())
-def ReplicationPad1dModule_basic(module, tu: TestUtils):
-    module.forward(tu.rand(1, 15, 20, low=-1))
-
-
 class ReplicationPad2dModule_right0_module(torch.nn.Module):
     def __init__(self):
         super().__init__()

projects/pt1/python/torch_mlir_e2e_test/test_suite/padding.py

@@ -13,6 +13,52 @@
 # ==============================================================================
 
 
+class ReplicationPad1dModule_3DInput(torch.nn.Module):
+    def __init__(self):
+        super().__init__()
+
+    @export
+    @annotate_args(
+        [
+            None,
+            ([-1, -1, -1], torch.float32, True),
+        ]
+    )
+    def forward(self, x):
+        return torch.ops.aten.replication_pad1d(x, [3, 5])
+
+
+@register_test_case(module_factory=lambda: ReplicationPad1dModule_3DInput())
+def ReplicationPad1dModule_3DInput_basic(module, tu: TestUtils):
+    module.forward(tu.rand(1, 15, 20, low=-1))
+
+
+# ==============================================================================
+
+
+class ReplicationPad1dModule_2DInput(torch.nn.Module):
+    def __init__(self):
+        super().__init__()
+
+    @export
+    @annotate_args(
+        [
+            None,
+            ([-1, -1], torch.float32, True),
+        ]
+    )
+    def forward(self, x):
+        return torch.ops.aten.replication_pad1d(x, [2, 3])
+
+
+@register_test_case(module_factory=lambda: ReplicationPad1dModule_2DInput())
+def ReplicationPad1dModule_2DInput_basic(module, tu: TestUtils):
+    module.forward(tu.rand(7, 12, low=-1))
+
+
+# ==============================================================================
+
+
 class ReflectionPad2dModule(torch.nn.Module):
     def __init__(self):
         super().__init__()