Skip to content

Commit a25355e

Browse files
cmdr2ggerganov
authored andcommitted
cpu: fix cpu backend's supports-op for GET_ROWS_BACK. fixes a fatal when running test-backend-ops with only the CPU backend (ggml/1190)
1 parent e959d32 commit a25355e

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

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

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -425,6 +425,8 @@ static bool ggml_backend_cpu_device_supports_op(ggml_backend_dev_t dev, const st
425425
}
426426
case GGML_OP_IM2COL_BACK:
427427
return src0->type == GGML_TYPE_F32 && src1->type == GGML_TYPE_F32;
428+
case GGML_OP_GET_ROWS_BACK:
429+
return src0->type == GGML_TYPE_F32 || src0->type == GGML_TYPE_F16;
428430
case GGML_OP_OUT_PROD:
429431
return (src0->type == GGML_TYPE_F32 || (ggml_is_quantized(src0->type) && src0->ne[2] == src1->ne[2] && src0->ne[3] == src1->ne[3])) &&
430432
src1->type == GGML_TYPE_F32 && op->type == GGML_TYPE_F32;

0 commit comments

Comments
 (0)