@@ -5998,6 +5998,14 @@ static std::vector<std::unique_ptr<test_case>> make_test_cases_eval() {
59985998    test_cases.emplace_back (new  test_sum ());
59995999    test_cases.emplace_back (new  test_sum_rows ());
60006000    test_cases.emplace_back (new  test_mean ());
6001+     test_cases.emplace_back (new  test_sum (GGML_TYPE_F32, { 33 , 1 , 1 , 1  }));
6002+     test_cases.emplace_back (new  test_sum_rows (GGML_TYPE_F32, { 33 , 1 , 1 , 1  }));
6003+     test_cases.emplace_back (new  test_mean (GGML_TYPE_F32, { 33 , 1 , 1 , 1  }));
6004+     test_cases.emplace_back (new  test_sum (GGML_TYPE_F32, { 33 , 1024 , 1 , 1  }));
6005+     test_cases.emplace_back (new  test_sum_rows (GGML_TYPE_F32, { 33 , 1024 , 1 , 1  }));
6006+     test_cases.emplace_back (new  test_sum (GGML_TYPE_F32, { 33 , 256 , 1 , 1  }));
6007+     test_cases.emplace_back (new  test_sum_rows (GGML_TYPE_F32, { 33 , 256 , 1 , 1  }));
6008+     test_cases.emplace_back (new  test_mean (GGML_TYPE_F32, { 33 , 256 , 1 , 1  }));
60016009    test_cases.emplace_back (new  test_group_norm (GGML_TYPE_F32, {64 , 64 , 320 , 1 }));
60026010    test_cases.emplace_back (new  test_group_norm (GGML_TYPE_F32, {9 , 9 , 1280 , 1 }));
60036011    test_cases.emplace_back (new  test_acc ());
@@ -6179,14 +6187,11 @@ static std::vector<std::unique_ptr<test_case>> make_test_cases_perf() {
61796187        test_cases.emplace_back (new  test_add_id (GGML_TYPE_F32, GGML_TYPE_F32, 2880 , 32 , 4 , n_token));
61806188    }
61816189
6182-     std::vector<std::array<int64_t , 4 >> reduce_rows_cases;
6183-     for  (int  j=0 ; j<14 ; ++j){
6184-         for  (int  i=7 ; i<14 ;++i){
6185-         {
6186-                 reduce_rows_cases.push_back ({ int64_t (std::pow (2 ,i)), int64_t (std::pow (2 ,j)), 1 , 1  });
6187-             }
6188-         }
6189-     }
6190+     std::vector<std::array<int64_t , 4 >> reduce_rows_cases = {
6191+         { 8192 , 1 ,    1 , 1  },
6192+         { 8192 , 8192 , 1 , 1  },
6193+         { 128 ,  8192 , 1 , 1  },
6194+     };
61906195
61916196    for  (auto  it: reduce_rows_cases){
61926197        test_cases.emplace_back (new  test_mean (GGML_TYPE_F32, it));
0 commit comments