added a few edge test cases
This commit is contained in:
parent
febee580c8
commit
f2187bbfa2
|
|
@ -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 s0 : { 1, 3 }) {
|
||||||
for (uint32_t p1 : { 2, 5 }) {
|
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 Cout : { 1, 12 }) {
|
||||||
for (uint32_t KH : { 1, 2, 3, 11 }) {
|
for (uint32_t KH : { 1, 2, 3, 11 }) {
|
||||||
for (uint32_t KW : { 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},
|
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));
|
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},
|
test_cases.emplace_back(new test_conv_2d( { 24, 24, 96, 1 }, { 3, 3, 96, 8},
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue