diff --git a/tests/test-backend-ops.cpp b/tests/test-backend-ops.cpp index 3f71cf1cf4..fb0224d480 100644 --- a/tests/test-backend-ops.cpp +++ b/tests/test-backend-ops.cpp @@ -5826,7 +5826,7 @@ static std::vector> make_test_cases_eval() { for (uint32_t s0 : { 1, 3 }) { for (uint32_t p1 : { 2, 5 }) { - for (uint32_t Cin : { 1, 16, 25, 48 }) { + for (uint32_t Cin : { 1, 25 }) { for (uint32_t Cout : { 1, 12 }) { for (uint32_t KH : { 1, 2, 3, 11 }) { for (uint32_t KW : { 1, 2, 3, 11 }) { @@ -5848,6 +5848,14 @@ static std::vector> make_test_cases_eval() { } } + test_cases.emplace_back(new test_conv_2d( { 16, 16, 8, 1}, { 3, 3, 8, 12}, + GGML_TYPE_F16, 1, 1, 1, 1, 1, 1, false)); + test_cases.emplace_back(new test_conv_2d( { 16, 16, 16, 1}, { 3, 3, 16, 6}, + GGML_TYPE_F16, 1, 1, 1, 1, 1, 1, false)); + test_cases.emplace_back(new test_conv_2d( { 16, 16, 24, 1}, { 3, 3, 24, 6}, + GGML_TYPE_F16, 1, 1, 1, 1, 1, 1, false)); + test_cases.emplace_back(new test_conv_2d( { 16, 16, 8, 3}, { 3, 3, 8, 6}, + GGML_TYPE_F16, 1, 1, 1, 1, 1, 1, false)); test_cases.emplace_back(new test_conv_2d( { 24, 24, 32, 1 }, { 3, 3, 32, 8}, GGML_TYPE_F16, 1, 1, 1, 1, 1, 1, false)); test_cases.emplace_back(new test_conv_2d( { 24, 24, 96, 1 }, { 3, 3, 96, 8},