diff --git a/ggml.c b/ggml.c
index f913c213e..be17e6dcd 100644
--- a/ggml.c
+++ b/ggml.c
@@ -5569,8 +5569,8 @@ struct ggml_tensor * ggml_pool_2d(
         int                   k1,
         int                   s0,
         int                   s1,
-        int                   p0,
-        int                   p1) {
+        float                 p0,
+        float                 p1) {
 
     bool is_node = false;
 
diff --git a/ggml.h b/ggml.h
index 5a56f73b7..cf568b3e5 100644
--- a/ggml.h
+++ b/ggml.h
@@ -1600,8 +1600,8 @@ extern "C" {
             int                   k1,
             int                   s0,
             int                   s1,
-            int                   p0,
-            int                   p1);
+            float                 p0,
+            float                 p1);
 
     // nearest interpolate
     // used in stable-diffusion