commit 334686bde752a8b34d02aac069cf3f910f7d8b70
author Ramin Azarmehr <razarmehr@apple.com> Fri Sep 30 22:57:57 2022 +0000
committer PyTorch MergeBot <pytorchmergebot@users.noreply.github.com> Fri Sep 30 22:57:57 2022 +0000
tree 40153b998332bcfb96b9fb0da1527ba1f973e568
parent 24fc680ee4228225c01fb6699210056ca2603a3f

Fix the dimension of padding to match the input's dimension (#85990)

Fixes #85143

Pull Request resolved: https://github.com/pytorch/pytorch/pull/85990
Approved by: https://github.com/malfet, https://github.com/kulinseth

test/test_mps.py

diff --git a/test/test_mps.py b/test/test_mps.py
index 6ccc6b1..aa3f1d0 100644
--- a/test/test_mps.py
+++ b/test/test_mps.py
@@ -3508,6 +3508,8 @@
         helper((2, 1, 6, 8), (2, 4, 3, 5), nn.ReplicationPad2d)
         # Constant Pad 2D
         helper((2, 1, 6, 8), (2, 4, 3, 5), nn.ConstantPad2d)
+        # input size < pad size
+        helper((1, 2, 3), (0, 0, 0, 1), nn.ConstantPad2d)
 
         # 3D Padding
         helper((2, 4, 6, 8, 4), (1, 3, 3, 5, 3, 4), nn.ReflectionPad3d)