Skip to content

Commit b9cc76d

Browse files
committed
ggml : fix ggml_backend_cpu_supports_op() for CPY (#0)
1 parent 7dbdba5 commit b9cc76d

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

ggml-backend.c

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -822,7 +822,11 @@ GGML_CALL static enum ggml_status ggml_backend_cpu_graph_compute(ggml_backend_t
822822
GGML_CALL static bool ggml_backend_cpu_supports_op(ggml_backend_t backend, const struct ggml_tensor * op) {
823823
switch (op->op) {
824824
case GGML_OP_CPY:
825-
return op->type != GGML_TYPE_IQ2_XXS && op->type != GGML_TYPE_IQ2_XS && op->type != GGML_TYPE_IQ1_S; // missing type_traits.from_float
825+
return
826+
op->type != GGML_TYPE_IQ2_XXS &&
827+
op->type != GGML_TYPE_IQ2_XS &&
828+
op->type != GGML_TYPE_IQ1_S &&
829+
op->type != GGML_TYPE_IQ1_M; // missing type_traits.from_float
826830
case GGML_OP_MUL_MAT:
827831
return op->src[1]->type == GGML_TYPE_F32 || op->src[1]->type == ggml_internal_get_type_traits(op->src[0]->type).vec_dot_type;
828832
default:

0 commit comments

Comments
 (0)