OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 12:20:52 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity

Add torchcheck for replication_pad3d_backward (#151986)

Browse Source 
Fixes #142833

Add check on channel dimension, logic same to the CUDA implementation 78bbb468c6/aten/src/ATen/native/cuda/ReplicationPadding.cu (L347)

Pull Request resolved: https://github.com/pytorch/pytorch/pull/151986
Approved by: https://github.com/mikaylagawarecki
This commit is contained in:
cz2h 2025-07-25 19:48:47 +00:00 committed by PyTorch MergeBot
parent 51cd6697cd
commit ee4c5c7cd2
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
aten/src/ATen/native/ReplicationPadding.cpp
View File

@ -229,17 +229,20 @@ void replication_pad3d_backward_out_cpu_template(
int pbottom = paddingSize[3];
int pfront = paddingSize[4];
int pback = paddingSize[5];
int dimc = 0;
int dimw = 3;
int dimh = 2;
int dimd = 1;
if (input.dim() == 5) {
dimc++;
dimw++;
dimh++;
dimd++;
}
/* sizes */
int64_t ichannel = input.size(dimc);
int64_t idepth = input.size(dimd);
int64_t iheight = input.size(dimh);
int64_t iwidth = input.size(dimw);
@ -249,6 +252,9 @@ void replication_pad3d_backward_out_cpu_template(
at::native::padding::check_valid_input<3>(input, paddingSize);
TORCH_CHECK(ichannel == gradOutput.size(dimc),
"gradOutput width unexpected. Expected: ", ichannel, ", Got: ",
gradOutput.size(dimc));
TORCH_CHECK(owidth == gradOutput.size(dimw),
"gradOutput width unexpected. Expected: ", owidth, ", Got: ",
gradOutput.size(dimw));