🐛 Describe the bug

Floating point exception in mkldnn_convolution when stride contains 0s.

Example to reproduce

import torch

self = torch.full((1, 1, 1, 24000,), 1, dtype=torch.float32, requires_grad=False)
weight = torch.full((1, 1, 1, 24000,), 1, dtype=torch.float32, requires_grad=False)
bias = torch.full((1, 1, 1, 24000,), 1, dtype=torch.float32, requires_grad=False)
padding = [0, 0]
stride = [0, 0]
dilation = [0, 0]
groups = 0
torch.mkldnn_convolution(self, weight, bias, padding, stride, dilation, groups)

Result

floating point exception

Expected behavior

Graceful termination or a RuntimeError to be thrown.

Note

This bug was discovered using fuzz testing.

Versions

PyTorch version: 1.9.0a0+git4d36e60
Is debug build: True
CUDA used to build PyTorch: None
ROCM used to build PyTorch: N/A

OS: Debian GNU/Linux 11 (bullseye) (x86_64)
GCC version: (Debian 10.2.1-6) 10.2.1 20210110
Clang version: 11.0.1-2
CMake version: version 3.22.1
Libc version: glibc-2.31

Python version: 3.9.7 (default, Sep 16 2021, 13:09:58) [GCC 7.5.0] (64-bit runtime)
Python platform: Linux-5.10.0-11-amd64-x86_64-with-glibc2.31
Is CUDA available: False
CUDA runtime version: No CUDA
GPU models and configuration: No CUDA
Nvidia driver version: No CUDA
cuDNN version: No CUDA
HIP runtime version: N/A
MIOpen runtime version: N/A

Versions of relevant libraries:
[pip3] numpy==1.22.2
[pip3] torch==1.9.0a0+git4d36e60
[conda] mkl 2022.0.1 h06a4308_117
[conda] mkl-include 2022.0.1 h06a4308_117
[conda] numpy 1.19.5 pypi_0 pypi
[conda] torch 1.9.0a0+git4d36e60 dev_0

cc @ezyang @gchanan @zou3519 @VitalyFedyunin @gujinghui @PenghuiCheng @XiaobingSuper @jianyuh