Skip to content

Commit 0006f5a

Browse files
authored
ggml : update ggml_backend_cpu_device_supports_op (ggml-org#10867)
* ggml : fix cpy op for IQ-quants to use reference impl ggml-ci * ggml : disable tests involving i-matrix quantization * ggml : update ggml_backend_cpu_device_supports_op ggml-ci
1 parent 05c3a44 commit 0006f5a

File tree

2 files changed

+5
-2
lines changed

2 files changed

+5
-2
lines changed

ggml/src/ggml-cpu/ggml-cpu.cpp

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -394,8 +394,11 @@ static bool ggml_backend_cpu_device_supports_op(ggml_backend_dev_t dev, const st
394394
switch (op->op) {
395395
case GGML_OP_CPY:
396396
return
397+
op->type != GGML_TYPE_IQ3_XXS &&
398+
op->type != GGML_TYPE_IQ3_S &&
397399
op->type != GGML_TYPE_IQ2_XXS &&
398400
op->type != GGML_TYPE_IQ2_XS &&
401+
op->type != GGML_TYPE_IQ2_S &&
399402
op->type != GGML_TYPE_IQ1_S &&
400403
op->type != GGML_TYPE_IQ1_M; // missing type_traits.from_float
401404
case GGML_OP_MUL_MAT:

tests/test-backend-ops.cpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3549,8 +3549,8 @@ static std::vector<std::unique_ptr<test_case>> make_test_cases_eval() {
35493549

35503550
for (ggml_type type_src : {GGML_TYPE_F16, GGML_TYPE_F32}) {
35513551
for (ggml_type type_dst : all_types) {
3552-
test_cases.emplace_back(new test_cpy(type_src, type_dst, {256, 4, 4, 4}));
3553-
test_cases.emplace_back(new test_cpy(type_src, type_dst, {256, 2, 3, 4}, {0, 2, 1, 3})); // cpy by rows
3552+
test_cases.emplace_back(new test_cpy(type_src, type_dst, {256, 4, 4, 4}));
3553+
test_cases.emplace_back(new test_cpy(type_src, type_dst, {256, 2, 3, 4}, {0, 2, 1, 3})); // cpy by rows
35543554
}
35553555
}
35563556
for (ggml_type type_src : {GGML_TYPE_F16, GGML_TYPE_F32}) {

0 commit comments

Comments
 (0)