From 89103a856c8437b668f75aae86fd736465bddfab Mon Sep 17 00:00:00 2001 From: bssrdf Date: Mon, 10 Nov 2025 15:26:09 -0500 Subject: [PATCH] updated test cases using tensor shapes from sd.cpp wan video generation --- tests/test-backend-ops.cpp | 19 ++++++++++--------- tests/test-conv3d.cpp | 9 +++++++-- 2 files changed, 17 insertions(+), 11 deletions(-) diff --git a/tests/test-backend-ops.cpp b/tests/test-backend-ops.cpp index f8a9b8f4b8..49a4061f07 100644 --- a/tests/test-backend-ops.cpp +++ b/tests/test-backend-ops.cpp @@ -7375,10 +7375,11 @@ static std::vector> make_test_cases_perf() { } - for (ggml_type kernel_type : {GGML_TYPE_F32, GGML_TYPE_F16}) { + // for (ggml_type kernel_type : {GGML_TYPE_F32, GGML_TYPE_F16}) { + for (ggml_type kernel_type : {GGML_TYPE_F16}) { for (int N : {1}) { - for (int IC : {320, 640}) { - for (int OC : {320, 640}) { + for (int IC : {48, 320, 640, 1024}) { + for (int OC : {320, 640, 1024, 2048}) { for (int s0 : {1}) { for (int p1 : {1}) { for (int d2 : {1}) { @@ -7404,12 +7405,12 @@ static std::vector> make_test_cases_perf() { OC, KD, KH, KW, s0, s1, s2, p0, p1, p2, d0, d1, d2, kernel_type)); - IW = 208; IH = 316; - test_cases.emplace_back(new test_conv_3d( - N, IC, ID, IH, IW, - OC, KD, KH, KW, - s0, s1, s2, p0, p1, p2, d0, d1, d2, - kernel_type)); + // IW = 208; IH = 316; + // test_cases.emplace_back(new test_conv_3d( + // N, IC, ID, IH, IW, + // OC, KD, KH, KW, + // s0, s1, s2, p0, p1, p2, d0, d1, d2, + // kernel_type)); } } } diff --git a/tests/test-conv3d.cpp b/tests/test-conv3d.cpp index e05bc50090..a34169c36d 100644 --- a/tests/test-conv3d.cpp +++ b/tests/test-conv3d.cpp @@ -358,7 +358,12 @@ int main(void) // std::make_tuple(320,1280,104,152,8,3,3,3), // std::make_tuple(1280,1280,104,152,8,3,3,3), // std::make_tuple(320,1280,208,304,4,3,3,3), - // std::make_tuple(640,1280,208,304,4,3,3,3), + std::make_tuple(1024,2048,30,52,3,3,3,3), + std::make_tuple(1024,2048,52,76,4,3,3,3), + std::make_tuple(1024,2048,52,76,6,3,3,3), + std::make_tuple(48,3072,64,64,9,2,2,1), + std::make_tuple(48,3072,64,64,17,2,2,1), + std::make_tuple(48,3072,64,64,33,2,2,1), }; int k = 0; @@ -380,7 +385,7 @@ int main(void) // fprintf(stderr, "%s: compute buffer size: %.2f MB\n", __func__, mem_size/1024.0f/1024.0f); - int iterations = 0; + int iterations = 20; double run_time0; std::vector im2col_data = compute_graph(model, allocr, build_graph_0, iterations,