added a few edge test cases

This commit is contained in:
bssrdf 2025-11-16 12:05:23 -05:00
parent febee580c8
commit f2187bbfa2
1 changed files with 9 additions and 1 deletions

View File

@ -5826,7 +5826,7 @@ static std::vector<std::unique_ptr<test_case>> 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<std::unique_ptr<test_case>> 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},