tests : fix im2col usage

diff --git a/tests/test-conv1d.cpp b/tests/test-conv1d.cpp
index 936f9661d2b71657660354340f38ca9cd9bf25ca..af0dd66757ddd5f2c3a5e48a4d73ede3049879e8 100644 (file)
--- 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 06398d0b82fa4d35517e80e68ea3ff89e4ce0cf7..b9e9e262a99ca5aaeb27f34cbade13b079214547 100644 (file)
--- 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);