test

microsoft · Feb 12, 2025 · 8cdd55d · 8cdd55d
1 parent 1c64678
commit 8cdd55d
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 19 deletions.
diff --git a/onnxscript/function_libs/torch_lib/ops/nn.py b/onnxscript/function_libs/torch_lib/ops/nn.py
@@ -1528,7 +1528,7 @@ def aten_pad(
         "circular": "wrap",
     }[mode]
 
-    return op.Pad(self, paddings, constant_value=const_value, mode=mode_mappings[mode])
+    return op.Pad(self, paddings, constant_value=const_value, mode=onnx_mode)
 
 
 def aten_pad_sequence(

diff --git a/tests/function_libs/torch_lib/ops_test_data.py b/tests/function_libs/torch_lib/ops_test_data.py
@@ -1082,26 +1082,14 @@ def _where_input_wrangler(
         input_wrangler=_nll_loss_input_wrangler,
         tolerance={torch.float16: (5e-2, 1e-2)},
     ),
-    TorchLibOpInfo("nn.functional.pad", nn_ops.aten_pad).skip(
-        variant_name="constant",
-        test_class_name="TestOutputConsistencyEager",
-        reason="fixme: Eager mode does not construct a correct signature for inputs of concat",
-    ).skip(
-        variant_name="reflect",
-        test_class_name="TestOutputConsistencyEager",
-        reason="fixme: Eager mode does not construct a correct signature for inputs of concat",
-    ).skip(
+    TorchLibOpInfo("nn.functional.pad", nn_ops.aten_pad)
+    .skip(
         variant_name="circular",
-        test_class_name="TestOutputConsistencyEager",
-        reason="fixme: Eager mode does not construct a correct signature for inputs of concat",
-    ).skip(
-        variant_name="replicate",
-        test_class_name="TestOutputConsistencyEager",
-        reason="fixme: Eager mode does not construct a correct signature for inputs of concat",
-    ).skip(
+        reason="fixme: ORT does not support the circular mode",
+    )
+    .skip(
         variant_name="replicate_negative",
-        test_class_name="TestOutputConsistencyEager",
-        reason="fixme: Eager mode does not construct a correct signature for inputs of concat",
+        reason="fixme: The implementation for negative paddings is not correct",
     ),
     TorchLibOpInfo(
         "nn.functional.pixel_shuffle",