From 66dd123b0f54bd1a6baaaa1c25c322c0132d2312 Mon Sep 17 00:00:00 2001 From: zhangjidong <1119708529@qq.com> Date: Wed, 31 Jan 2024 10:22:09 +0800 Subject: [PATCH] pool2d float padding fallback --- ggml.c | 4 ++-- ggml.h | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) 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