@@ -37,8 +37,8 @@ struct test_model {
3737
3838void  load_model (test_model & model, bool  use_gpu = false ) {
3939    //  create data
40-     int  KW = 3 , KH = 3 , IC = 256 , OC = 256 ;
41-     int  IW = 832 , IH = 1216 , N = 1 ;
40+     int  KW = 3 , KH = 3 , IC = 128 , OC = 128 ;
41+     int  IW = 64 , IH = 96 , N = 1 ;
4242
4343    printf ("  input: IC = %d, OC = %d, IW = %d, IH = %d \n  "  , IC, OC, IW, IH);
4444
@@ -382,15 +382,15 @@ int main(void)
382382
383383    bool  passed = true ;
384384    //  for(int i = 0; i < ggml_nelements(wino_res); i++) {
385-     //   for(int i = 0; i < 3*28; i++) {
386-     //       float diff = fabs(conv2d_data[i] - wino_data[i]);
387-     //       // if(diff > 1.e-4) {
388-     //             printf("(%f, %f, %f, %d) \n", 
389-     //             conv2d_data[i],
390-     //             wino_data[i], diff, i);
391-     //           // break;
392-     //       // }
393-     //   }
385+     for (int  i = 0 ; i < 3 *28 ; i++) {
386+         float  diff = fabs (conv2d_data[i] - wino_data[i]);
387+         //  if(diff > 1.e-4) {
388+               printf (" (%f, %f, %f, %d) \n "  , 
389+               conv2d_data[i],
390+               wino_data[i], diff, i);
391+             //  break;
392+         //  }
393+     }
394394
395395
396396
0 commit comments