fix div 0 error in conv1/2/3

paddle/phi/kernels/impl/conv_kernel_impl.h

@@ -124,6 +124,11 @@ void ConvKernelImpl(const Context& dev_ctx,
                               filter.numel() / filter.dims()[0]};
   filter.Resize(filter_matrix_shape);
 
+  PADDLE_ENFORCE_NE(
+      transformed_output.dims()[0] * transformed_output.dims()[1],
+      0,
+      phi::errors::InvalidArgument("The Output size should not be 0."));
+
   DDim output_matrix_shape = {
       transformed_output.dims()[1],
       transformed_output.numel() /

python/paddle/fluid/tests/unittests/test_functional_conv1d.py

@@ -70,5 +70,17 @@ def setUp(self):
         self.data_format = "NCL"
 
 
+class TestFunctionalConv1DErrorCase2(TestFunctionalConv1DError):
+    def setUp(self):
+        self.input = np.random.randn(0, 0, 0)
+        self.filter = np.random.randn(1, 0, 0)
+        self.bias = None
+        self.padding = 0
+        self.stride = 1
+        self.dilation = 1
+        self.groups = 1
+        self.data_format = "NCL"
+
+
 if __name__ == "__main__":
     unittest.main()

python/paddle/fluid/tests/unittests/test_functional_conv2d.py

@@ -570,6 +570,20 @@ def setUp(self):
         self.data_format = "NCHW"
 
 
+class TestFunctionalConv2DErrorCase14(TestFunctionalConv2DErrorCase12):
+    def setUp(self):
+        self.input = np.random.randn(0, 0, 0, 0)
+        self.filter = np.random.randn(1, 0, 0, 0)
+        self.num_filters = 0
+        self.filter_size = 0
+        self.bias = None
+        self.padding = 0
+        self.stride = 1
+        self.dilation = 1
+        self.groups = 1
+        self.data_format = "NCHW"
+
+
 if __name__ == "__main__":
     paddle.enable_static()
     unittest.main()

python/paddle/fluid/tests/unittests/test_functional_conv3d.py

@@ -545,6 +545,20 @@ def setUp(self):
         self.data_format = "NCDHW"
 
 
+class TestFunctionalConv3DErrorCase13(TestFunctionalConv3DErrorCase11):
+    def setUp(self):
+        self.input = np.random.randn(0, 0, 0, 0, 0)
+        self.filter = np.random.randn(1, 0, 0, 0, 0)
+        self.num_filters = 1
+        self.filter_size = 1
+        self.bias = None
+        self.padding = 0
+        self.stride = 1
+        self.dilation = 1
+        self.groups = 1
+        self.data_format = "NCDHW"
+
+
 if __name__ == "__main__":
     paddle.enable_static()
     unittest.main()