From: Georgi Gerganov <redacted>
Date: Sat, 10 Feb 2024 07:45:40 +0000 (+0200)
Subject: tests : fix im2col usage
X-Git-Tag: upstream/0.0.1642~981
X-Git-Url: https://git.djapps.eu/?a=commitdiff_plain;h=3c3270160015cf0dbe2a5290870798dd5c6acc6d;p=pkg%2Fggml%2Fsources%2Fggml

tests : fix im2col usage
---

diff --git a/tests/test-conv1d.cpp b/tests/test-conv1d.cpp
index 936f9661..af0dd667 100644
--- a/tests/test-conv1d.cpp
+++ b/tests/test-conv1d.cpp
@@ -159,7 +159,7 @@ struct ggml_cgraph * build_graph(const test_model& model, struct ggml_allocr * a
     int d0 = 1;
 
     // split conv1d in fundamental methods for test unit
-    struct ggml_tensor* im2col_0 = ggml_im2col(ctx0, model.a, model.b, s0, 0, p0, 0, d0, 0, false);
+    struct ggml_tensor* im2col_0 = ggml_im2col(ctx0, model.a, model.b, s0, 0, p0, 0, d0, 0, false, GGML_TYPE_F16);
     ggml_set_name(im2col_0, "im2col_res");
     ggml_build_forward_expand(gf, im2col_0);
 
diff --git a/tests/test-conv2d.cpp b/tests/test-conv2d.cpp
index 06398d0b..b9e9e262 100644
--- a/tests/test-conv2d.cpp
+++ b/tests/test-conv2d.cpp
@@ -162,7 +162,7 @@ struct ggml_cgraph * build_graph(const test_model& model, struct ggml_allocr * a
     int d1 = 1;
 
     // split conv2d in fundamental methods for test unit
-    struct ggml_tensor* im2col_0 = ggml_im2col(ctx0, model.a, model.b, s0, s1, p0, p1, d0, d1, true);
+    struct ggml_tensor* im2col_0 = ggml_im2col(ctx0, model.a, model.b, s0, s1, p0, p1, d0, d1, true, GGML_TYPE_F16);
     ggml_set_name(im2col_0, "im2col_res");
     ggml_build_forward_expand(gf, im2col_0);