Skip to content

Commit a3e78dc

Browse files
committed
tests : add cpy tests for all types
ggml-ci
1 parent b971d06 commit a3e78dc

File tree

1 file changed

+8
-7
lines changed

1 file changed

+8
-7
lines changed

tests/test-backend-ops.cpp

Lines changed: 8 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -3929,14 +3929,15 @@ static std::vector<std::unique_ptr<test_case>> make_test_cases_eval() {
39293929
test_cases.emplace_back(new test_set(GGML_TYPE_I32, GGML_TYPE_I32, {6, 5, 4, 3}, dim));
39303930
}
39313931

3932-
test_cases.emplace_back(new test_cpy(GGML_TYPE_Q8_0, GGML_TYPE_Q8_0, {256, 2, 3, 4}, {0, 1, 2, 3}));
3933-
test_cases.emplace_back(new test_cpy(GGML_TYPE_Q8_0, GGML_TYPE_Q8_0, {256, 2, 3, 4}, {0, 2, 1, 3}));
3932+
for (int nr = 1; nr < 4; ++nr) {
3933+
for (ggml_type type : all_types) {
3934+
const auto neb = ggml_blck_size(type);
39343935

3935-
test_cases.emplace_back(new test_cpy(GGML_TYPE_Q4_0, GGML_TYPE_Q4_0, {256, 2, 3, 4}, {0, 1, 2, 3}));
3936-
test_cases.emplace_back(new test_cpy(GGML_TYPE_Q4_0, GGML_TYPE_Q4_0, {256, 2, 3, 4}, {0, 2, 3, 1}));
3937-
3938-
test_cases.emplace_back(new test_cpy(GGML_TYPE_Q4_K, GGML_TYPE_Q4_K, {256, 2, 3, 4}, {0, 1, 2, 3}));
3939-
test_cases.emplace_back(new test_cpy(GGML_TYPE_Q4_K, GGML_TYPE_Q4_K, {256, 2, 3, 4}, {0, 3, 1, 2}));
3936+
test_cases.emplace_back(new test_cpy(type, type, {4*neb, 2, 3, 4}, {0, 1, 2, 3}));
3937+
test_cases.emplace_back(new test_cpy(type, type, {4*neb, 2, 3, 4}, {0, 2, 1, 3}));
3938+
test_cases.emplace_back(new test_cpy(type, type, {4*neb, 2, 3, 4}, {0, 3, 1, 2}));
3939+
}
3940+
}
39403941

39413942
for (ggml_type type_src : {GGML_TYPE_F16, GGML_TYPE_F32}) {
39423943
for (ggml_type type_dst : all_types) {

0 commit comments

Comments
 (0)