fix test case

ggml-org · Dec 25, 2024 · 864e4a7 · 864e4a7
1 parent 1d606ed
commit 864e4a7
Showing 1 changed file with 11 additions and 11 deletions.
diff --git a/tests/test-conv2d-winograd.cpp b/tests/test-conv2d-winograd.cpp
@@ -37,8 +37,8 @@ struct test_model {
 
 void load_model(test_model & model, bool use_gpu = false) {
     // create data
-    int KW = 3, KH = 3, IC = 256, OC = 256;
-    int IW = 832, IH = 1216, N = 1;
+    int KW = 3, KH = 3, IC = 128, OC = 128;
+    int IW = 64, IH = 96, N = 1;
 
     printf(" input: IC = %d, OC = %d, IW = %d, IH = %d \n ", IC, OC, IW, IH);
 
@@ -382,15 +382,15 @@ int main(void)
 
     bool passed = true;
     // for(int i = 0; i < ggml_nelements(wino_res); i++) {
-    // for(int i = 0; i < 3*28; i++) {
-    //     float diff = fabs(conv2d_data[i] - wino_data[i]);
-    //     // if(diff > 1.e-4) {
-    //           printf("(%f, %f, %f, %d) \n", 
-    //           conv2d_data[i],
-    //           wino_data[i], diff, i);
-    //         // break;
-    //     // }
-    // }
+    for(int i = 0; i < 3*28; i++) {
+        float diff = fabs(conv2d_data[i] - wino_data[i]);
+        // if(diff > 1.e-4) {
+              printf("(%f, %f, %f, %d) \n", 
+              conv2d_data[i],
+              wino_data[i], diff, i);
+            // break;
+        // }
+    }