Skip to content

Commit 77f2e5f

Browse files
committed
Modify perf and unit-testsz
1 parent 7b61f26 commit 77f2e5f

File tree

1 file changed

+13
-8
lines changed

1 file changed

+13
-8
lines changed

tests/test-backend-ops.cpp

Lines changed: 13 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -5813,6 +5813,14 @@ static std::vector<std::unique_ptr<test_case>> make_test_cases_eval() {
58135813
test_cases.emplace_back(new test_sum());
58145814
test_cases.emplace_back(new test_sum_rows());
58155815
test_cases.emplace_back(new test_mean());
5816+
test_cases.emplace_back(new test_sum(GGML_TYPE_F32, { 33, 1, 1, 1 }));
5817+
test_cases.emplace_back(new test_sum_rows(GGML_TYPE_F32, { 33, 1, 1, 1 }));
5818+
test_cases.emplace_back(new test_mean(GGML_TYPE_F32, { 33, 1, 1, 1 }));
5819+
test_cases.emplace_back(new test_sum(GGML_TYPE_F32, { 33, 1024, 1, 1 }));
5820+
test_cases.emplace_back(new test_sum_rows(GGML_TYPE_F32, { 33, 1024, 1, 1 }));
5821+
test_cases.emplace_back(new test_sum(GGML_TYPE_F32, { 33, 256, 1, 1 }));
5822+
test_cases.emplace_back(new test_sum_rows(GGML_TYPE_F32, { 33, 256, 1, 1 }));
5823+
test_cases.emplace_back(new test_mean(GGML_TYPE_F32, { 33, 256, 1, 1 }));
58165824
test_cases.emplace_back(new test_group_norm(GGML_TYPE_F32, {64, 64, 320, 1}));
58175825
test_cases.emplace_back(new test_group_norm(GGML_TYPE_F32, {9, 9, 1280, 1}));
58185826
test_cases.emplace_back(new test_acc());
@@ -5986,14 +5994,11 @@ static std::vector<std::unique_ptr<test_case>> make_test_cases_perf() {
59865994

59875995
test_cases.emplace_back(new test_mean(GGML_TYPE_F32, {256, 256, 3, 1}));
59885996

5989-
std::vector<std::array<int64_t, 4>> reduce_rows_cases;
5990-
for (int j=0; j<14; ++j){
5991-
for (int i=7; i<14;++i){
5992-
{
5993-
reduce_rows_cases.push_back({ int64_t(std::pow(2,i)), int64_t(std::pow(2,j)), 1, 1 });
5994-
}
5995-
}
5996-
}
5997+
std::vector<std::array<int64_t, 4>> reduce_rows_cases = {
5998+
{ 8192, 1, 1, 1 },
5999+
{ 8192, 8192, 1, 1 },
6000+
{ 128, 8192, 1, 1 },
6001+
};
59976002

59986003
for (auto it: reduce_rows_cases){
59996004
test_cases.emplace_back(new test_mean(GGML_TYPE_F32, it));

0 commit comments

Comments
 (0)