Skip to content

Commit fa9e415

Browse files
committed
minor update of test case
1 parent f95664c commit fa9e415

File tree

1 file changed

+21
-20
lines changed

1 file changed

+21
-20
lines changed

tests/test-conv2d.cpp

Lines changed: 21 additions & 20 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)