Skip to content

Commit 1da9854

Browse files
committed
reduce the batch size
1 parent 96e2320 commit 1da9854

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

tests/test-backend-ops.cpp

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -6564,8 +6564,7 @@ static std::vector<std::unique_ptr<test_case>> make_test_cases_eval() {
65646564
test_cases.emplace_back(new test_mul_mat(type_a, type_b, 16, 16, 1024, {3, 2}, {1, 1}));
65656565

65666566
// test cases with large batch size
6567-
test_cases.emplace_back(new test_mul_mat(type_a, type_b, 16, 8, 256, {1024, 2}, {1, 1}));
6568-
test_cases.emplace_back(new test_mul_mat(type_a, type_b, 16, 8, 256, {4096, 1}, {1, 1}));
6567+
test_cases.emplace_back(new test_mul_mat(type_a, type_b, 16, 8, 256, {1536, 1}, {1, 1}));
65696568
}
65706569
}
65716570
for (ggml_type type_a : other_types) {

0 commit comments

Comments
 (0)