Skip to content

Commit 694b5d1

Browse files
zhiyuan1iMollySophia
authored andcommitted
update tests for 1b6 3b 7b
1 parent 9b06a0e commit 694b5d1

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
@@ -4041,6 +4041,9 @@ static std::vector<std::unique_ptr<test_case>> make_test_cases_eval() {
40414041
test_cases.emplace_back(new test_rwkv_wkv6(GGML_TYPE_F32, 32, 64, 32, 1));
40424042
test_cases.emplace_back(new test_rwkv_wkv6(GGML_TYPE_F32, 32, 64, 32, 4));
40434043
test_cases.emplace_back(new test_rwkv_wkv6(GGML_TYPE_F32, 32, 64, 128, 4));
4044+
test_cases.emplace_back(new test_rwkv_wkv6(GGML_TYPE_F32, 32, 64, 128, 128));
4045+
test_cases.emplace_back(new test_rwkv_wkv6(GGML_TYPE_F32, 40, 64, 128, 128));
4046+
test_cases.emplace_back(new test_rwkv_wkv6(GGML_TYPE_F32, 64, 64, 128, 128));
40444047

40454048
test_cases.emplace_back(new test_rwkv_wkv7(GGML_TYPE_F32, 32, 64, 1, 1));
40464049
test_cases.emplace_back(new test_rwkv_wkv7(GGML_TYPE_F32, 32, 64, 32, 1));

0 commit comments

Comments
 (0)