frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix meta for constant_pad_nd (#159878)

Browse Source 
Fixes https://github.com/pytorch/pytorch/issues/144187

Pull Request resolved: https://github.com/pytorch/pytorch/pull/159878
Approved by: https://github.com/Skylion007, https://github.com/ezyang
This commit is contained in:
Isuru Fernando
2025-08-14 11:44:56 +00:00
committed by PyTorch MergeBot
parent e4de93f6a3
commit 781e9a7724
2 changed files with 22 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
torch/_meta_registrations.py
View File

@ -7638,6 +7638,20 @@ def _constant_pad_nd_meta(input, pad, value=0):
f"{l_inp} dimensions.",
)
if all(isinstance(p, utils.IntWithoutSymInt) and p <= 0 for p in pad):
c_input = input
for i in range(l_diff, l_inp):
pad_idx = 2 * (l_inp - i - 1)
if pad[pad_idx] < 0:
c_input = c_input.narrow(
i, -pad[pad_idx], c_input.shape[i] + pad[pad_idx]
)
if pad[pad_idx + 1] < 0:
c_input = c_input.narrow(i, 0, c_input.shape[i] + pad[pad_idx + 1])
return c_input.clone()
new_shape = list(input_sizes[:l_diff])
for i in range(l_pad):
pad_idx = len(pad) - ((i + 1) * 2)