add stride check for PoolOutputSize

PaddlePaddle · Feb 1, 2023 · 48ab823 · 48ab823
1 parent fd5b8ee
commit 48ab823
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/paddle/phi/kernels/funcs/pooling.h b/paddle/phi/kernels/funcs/pooling.h
@@ -371,6 +371,13 @@ inline int PoolOutputSize(int input_size,
                           int padding_2,
                           int stride,
                           bool ceil_mode) {
+  PADDLE_ENFORCE_NE(
+      stride,
+      0,
+      phi::errors::InvalidArgument(
+          "The stride of PoolOutputSize shall not be 0, but received %d.",
+          stride));
+
   int output_size;
   if (!ceil_mode) {
     output_size =