Skip to content

Commit 0468b99

Browse files
committed
test: More verbose printing and better cumsum tests
Branch: Mamba2SSD Signed-off-by: Gabe Goodhart <ghart@us.ibm.com>
1 parent e5587cb commit 0468b99

File tree

1 file changed

+8
-8
lines changed

1 file changed

+8
-8
lines changed

tests/test-backend-ops.cpp

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -1382,11 +1382,12 @@ struct test_case {
13821382
double err = nmse(f1.data(), f2.data(), f1.size());
13831383
if (err > ud->max_err) {
13841384
printf("[%s] NMSE = %.9f > %.9f ", ggml_op_desc(t1), err, ud->max_err);
1385-
//for (int i = 0; i < (int) f1.size(); i++) {
1386-
// printf("%5d %9.6f %9.6f, diff = %9.6f\n", i, f1[i], f2[i], f1[i] - f2[i]);
1387-
//}
1388-
//printf("\n");
1389-
//exit(1);
1385+
if (ud->verbose) {
1386+
for (int i = 0; i < (int) f1.size(); i++) {
1387+
printf("%5d %9.6f %9.6f, diff = %9.6f\n", i, f1[i], f2[i], f1[i] - f2[i]);
1388+
}
1389+
printf("\n");
1390+
}
13901391
ud->ok = false;
13911392
}
13921393
return true;
@@ -6938,8 +6939,7 @@ static std::vector<std::unique_ptr<test_case>> make_test_cases_eval(int verbose
69386939
test_cases.emplace_back(new test_cumsum(GGML_TYPE_F32, { 4, 2, 2, 1 }));
69396940
test_cases.emplace_back(new test_cumsum(GGML_TYPE_F16, { 4, 2, 2, 1 }));
69406941
test_cases.emplace_back(new test_cumsum(GGML_TYPE_BF16, { 4, 2, 2, 1 }));
6941-
test_cases.emplace_back(new test_cumsum(GGML_TYPE_F32, { 1024, 4, 2, 1 }));
6942-
// test_cases.emplace_back(new test_cumsum(GGML_TYPE_F32, { 2025, 5, 6, 3 }));
6942+
test_cases.emplace_back(new test_cumsum(GGML_TYPE_F32, { 2025, 5, 6, 3 }));
69436943

69446944
test_cases.emplace_back(new test_tri(GGML_TRI_TYPE_LOWER));
69456945
test_cases.emplace_back(new test_tri(GGML_TRI_TYPE_LOWER_DIAG));
@@ -7111,7 +7111,7 @@ static std::vector<std::unique_ptr<test_case>> make_test_cases_perf() {
71117111
test_cases.emplace_back(new test_cumsum(GGML_TYPE_F32, { 4, 2, 2, 1 }));
71127112
test_cases.emplace_back(new test_cumsum(GGML_TYPE_F16, { 4, 2, 2, 1 }));
71137113
test_cases.emplace_back(new test_cumsum(GGML_TYPE_BF16, { 4, 2, 2, 1 }));
7114-
test_cases.emplace_back(new test_cumsum(GGML_TYPE_F32, { 1024, 15, 26, 12 }));
7114+
test_cases.emplace_back(new test_cumsum(GGML_TYPE_F32, { 2025, 5, 6, 3 }));
71157115

71167116
test_cases.emplace_back(new test_tri(GGML_TRI_TYPE_LOWER));
71177117
test_cases.emplace_back(new test_tri(GGML_TRI_TYPE_LOWER_DIAG));

0 commit comments

Comments
 (0)