Skip to content

Commit 757ebb2

Browse files
zhiyuan1iMollySophia
authored andcommitted
update tests for 1b6 3b 7b
1 parent 8128d45 commit 757ebb2

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

tests/test-backend-ops.cpp

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -4059,6 +4059,9 @@ static std::vector<std::unique_ptr<test_case>> make_test_cases_eval() {
40594059
test_cases.emplace_back(new test_rwkv_wkv6(GGML_TYPE_F32, 32, 64, 32, 1));
40604060
test_cases.emplace_back(new test_rwkv_wkv6(GGML_TYPE_F32, 32, 64, 32, 4));
40614061
test_cases.emplace_back(new test_rwkv_wkv6(GGML_TYPE_F32, 32, 64, 128, 4));
4062+
test_cases.emplace_back(new test_rwkv_wkv6(GGML_TYPE_F32, 32, 64, 128, 128));
4063+
test_cases.emplace_back(new test_rwkv_wkv6(GGML_TYPE_F32, 40, 64, 128, 128));
4064+
test_cases.emplace_back(new test_rwkv_wkv6(GGML_TYPE_F32, 64, 64, 128, 128));
40624065

40634066
test_cases.emplace_back(new test_rwkv_wkv7(GGML_TYPE_F32, 32, 64, 1, 1));
40644067
test_cases.emplace_back(new test_rwkv_wkv7(GGML_TYPE_F32, 32, 64, 32, 1));

0 commit comments

Comments
 (0)