@@ -306,26 +306,27 @@ int main(void)
306306{
307307 ggml_time_init ();
308308 std::vector<std::tuple<int , int , int , int , int , int >> configs = {
309- // std::make_tuple(64,64,48,64,3,3),
310- // std::make_tuple(320,320,104,152,3,3),
311- // std::make_tuple(640,640,52,76,3,3),
312- // std::make_tuple(640,640,104,152,3,3),
313- // std::make_tuple(960,320,104,152,3,3),
314- // std::make_tuple(1280,1280,26,38,3,3),
315- // std::make_tuple(1280,1280,26,38,1,1),
316- // std::make_tuple(256,128,768,1024,3,3),
317- // std::make_tuple(128,3,768,1024,3,3),
318- // std::make_tuple(256,128,768,1024,1,1),
319- // std::make_tuple(512,256,384,512,1,1),
320- // std::make_tuple(1280,640,52,76,3,3),
321- // std::make_tuple(1920,1280,26,38,3,3),
322- // std::make_tuple(2560,1280,26,38,3,3),
309+ std::make_tuple (64 ,64 ,48 ,64 ,3 ,3 ),
310+ std::make_tuple (320 ,320 ,104 ,152 ,3 ,3 ),
311+ std::make_tuple (640 ,640 ,52 ,76 ,3 ,3 ),
312+ std::make_tuple (640 ,640 ,104 ,152 ,3 ,3 ),
313+ std::make_tuple (960 ,320 ,104 ,152 ,3 ,3 ),
314+ std::make_tuple (1280 ,1280 ,26 ,38 ,3 ,3 ),
323315 std::make_tuple (320 ,1280 ,26 ,38 ,3 ,3 ),
324- // std::make_tuple(512,512,104,152,3,3),
325- // std::make_tuple(512,512,208,304,3,3),
326- // std::make_tuple(512,256,416,608,3,3),
327- // std::make_tuple(256,128,832,1216,3,3),
328- // std::make_tuple(256,256,832,1216,3,3),
316+ std::make_tuple (1280 ,1280 ,26 ,38 ,1 ,1 ),
317+ std::make_tuple (256 ,128 ,768 ,1024 ,3 ,3 ),
318+ std::make_tuple (128 ,3 ,768 ,1024 ,3 ,3 ),
319+ std::make_tuple (256 ,128 ,768 ,1024 ,1 ,1 ),
320+ std::make_tuple (512 ,256 ,384 ,512 ,1 ,1 ),
321+ std::make_tuple (1280 ,640 ,52 ,76 ,3 ,3 ),
322+ std::make_tuple (1920 ,1280 ,26 ,38 ,3 ,3 ),
323+ std::make_tuple (2560 ,1280 ,26 ,38 ,3 ,3 ),
324+ std::make_tuple (320 ,1280 ,26 ,38 ,3 ,3 ),
325+ std::make_tuple (512 ,512 ,104 ,152 ,3 ,3 ),
326+ std::make_tuple (512 ,512 ,208 ,304 ,3 ,3 ),
327+ std::make_tuple (512 ,256 ,416 ,608 ,3 ,3 ),
328+ std::make_tuple (256 ,128 ,832 ,1216 ,3 ,3 ),
329+ std::make_tuple (256 ,256 ,832 ,1216 ,3 ,3 ),
329330 // std::make_tuple(320,256,1024,1920)
330331 };
331332
@@ -377,7 +378,7 @@ int main(void)
377378
378379 if (k==0 ) {
379380 k = 1 ;
380- fprintf (stderr, " | (IC, OC, IW, IH) | im2col+GEMM TIME | im2col+GEMM VRAM | implicit GEMM TIME | implicit GEMM VRAM \n " );
381+ fprintf (stderr, " | (IC, OC, IW, IH, KW, KH ) | im2col+GEMM TIME | im2col+GEMM VRAM | implicit GEMM TIME | implicit GEMM VRAM \n " );
381382 fprintf (stderr, " | --- | --- | --- | --- | --- \n " );
382383 }
383384
0 commit comments