Skip to content

Commit 3e81246

Browse files
authored
fix: correct ggml_pad_ext (#1133)
1 parent 98916e8 commit 3e81246

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

ggml_extend.hpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1033,7 +1033,7 @@ __STATIC_INLINE__ struct ggml_tensor* ggml_ext_pad(struct ggml_context* ctx,
10331033
int p3 = 0,
10341034
bool circular_x = false,
10351035
bool circular_y = false) {
1036-
return ggml_ext_pad_ext(ctx, x, p0, p0, p1, p1, p2, p2, p3, p3, circular_x, circular_y);
1036+
return ggml_ext_pad_ext(ctx, x, 0, p0, 0, p1, 0, p2, 0, p3, circular_x, circular_y);
10371037
}
10381038

10391039
// w: [OC,IC, KH, KW]
@@ -1062,7 +1062,7 @@ __STATIC_INLINE__ struct ggml_tensor* ggml_ext_conv_2d(struct ggml_context* ctx,
10621062
}
10631063

10641064
if ((p0 != 0 || p1 != 0) && (circular_x || circular_y)) {
1065-
x = ggml_ext_pad(ctx, x, p0, p1, 0, 0, circular_x, circular_y);
1065+
x = ggml_ext_pad_ext(ctx, x, p0, p0, p1, p1, 0, 0, 0, 0, circular_x, circular_y);
10661066
p0 = 0;
10671067
p1 = 0;
10681068
}

0 commit comments

Comments
 (0)