@@ -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