Skip to content

Commit e707662

Browse files
committed
add backend ops test
1 parent c2cbb16 commit e707662

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

tests/test-backend-ops.cpp

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -6562,6 +6562,10 @@ static std::vector<std::unique_ptr<test_case>> make_test_cases_eval() {
65626562
test_cases.emplace_back(new test_mul_mat(type_a, type_b, 16, 1, 1024, {3, 2}, {1, 1}));
65636563
test_cases.emplace_back(new test_mul_mat(type_a, type_b, 16, 8, 1024, {3, 2}, {1, 1}));
65646564
test_cases.emplace_back(new test_mul_mat(type_a, type_b, 16, 16, 1024, {3, 2}, {1, 1}));
6565+
6566+
// 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}));
65656569
}
65666570
}
65676571
for (ggml_type type_a : other_types) {

0 commit comments

Comments
 (0)