@@ -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