mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 12:54:11 +08:00
## Fixes https://github.com/pytorch/pytorch/issues/159462 also fixes #163569 , #163604 ## summary the issue is caused by the wrong stride of conv1d's result generated by meta_conv:4d5b3f2d5a/torch/_meta_registrations.py (L2453-L2471)
and the wrong stride will be used to codegen size assert in inductor:4d5b3f2d5a/torch/_inductor/ir.py (L6152-L6163)
## reason So why the computed stride is wrong in the meta_conv function? because the corresponding backend will convert conv1d to conv2d and change the input tensor' size and memory_format(channel last). but the meta_conv do not do this transformation, so a mismatch happend.4d5b3f2d5a/aten/src/ATen/native/Convolution.cpp (L1502-L1510)
just add corresponding logic in meta_conv. Pull Request resolved: https://github.com/pytorch/pytorch/pull/160408 Approved by: https://github.com/eellison, https://github.com/jansel, https://github.com/mlazos