diff --git a/.ci/docker/ci_commit_pins/pytorch.txt b/.ci/docker/ci_commit_pins/pytorch.txt index e3a53c8bcb5..0ebcdd3813c 100644 --- a/.ci/docker/ci_commit_pins/pytorch.txt +++ b/.ci/docker/ci_commit_pins/pytorch.txt @@ -1 +1 @@ -53a2908a10f414a2f85caa06703a26a40e873869 +gh/benjaminglass1/106/orig diff --git a/backends/apple/mps/test/test_mps.py b/backends/apple/mps/test/test_mps.py index a981d7ab8eb..75afaad0975 100644 --- a/backends/apple/mps/test/test_mps.py +++ b/backends/apple/mps/test/test_mps.py @@ -1522,7 +1522,7 @@ def test_mps_backend_constant_pad_nd(self): class PadModule(torch.nn.Module): def __init__(self): super().__init__() - self.constant_pad = torch.nn.ConstantPad2d((1, 2), 0) + self.constant_pad = torch.nn.ConstantPad2d((1, 2, 0, 0), 0) def forward(self, x): return self.constant_pad(x) diff --git a/backends/xnnpack/test/ops/test_static_constant_pad.py b/backends/xnnpack/test/ops/test_static_constant_pad.py index 53224071edd..c09cd8d46f6 100644 --- a/backends/xnnpack/test/ops/test_static_constant_pad.py +++ b/backends/xnnpack/test/ops/test_static_constant_pad.py @@ -105,7 +105,7 @@ def forward(self, x, y, z): class StaticConstantPad2d(torch.nn.Module): def __init__(self): super().__init__() - self.pad = torch.nn.ConstantPad2d([1, 2, 3, 4], 2.3) + self.pad = torch.nn.ConstantPad2d((1, 2, 3, 4), 2.3) def forward(self, x): y = self.pad(x)