From: cmdr2 <redacted>
Date: Fri, 11 Apr 2025 06:44:19 +0000 (+0530)
Subject: cpu: fix cpu backend's supports-op for GET_ROWS_BACK. fixes a fatal when running... 
X-Git-Tag: upstream/0.0.1982~40
X-Git-Url: https://git.djapps.eu/?a=commitdiff_plain;h=f71d538ece3fb32a04824dc6d1e73e360be9d22f;p=pkg%2Fggml%2Fsources%2Fggml

cpu: fix cpu backend's supports-op for GET_ROWS_BACK. fixes a fatal when running test-backend-ops with only the CPU backend (#1190)
---

diff --git a/src/ggml-cpu/ggml-cpu.cpp b/src/ggml-cpu/ggml-cpu.cpp
index 09f8382b..4b688a67 100644
--- a/src/ggml-cpu/ggml-cpu.cpp
+++ b/src/ggml-cpu/ggml-cpu.cpp
@@ -425,6 +425,8 @@ static bool ggml_backend_cpu_device_supports_op(ggml_backend_dev_t dev, const st
         }
         case GGML_OP_IM2COL_BACK:
             return src0->type == GGML_TYPE_F32 && src1->type == GGML_TYPE_F32;
+        case GGML_OP_GET_ROWS_BACK:
+            return src0->type == GGML_TYPE_F32 || src0->type == GGML_TYPE_F16;
         case GGML_OP_OUT_PROD:
             return (src0->type == GGML_TYPE_F32 || (ggml_is_quantized(src0->type) && src0->ne[2] == src1->ne[2] && src0->ne[3] == src1->ne[3])) &&
                 src1->type == GGML_TYPE_F32 && op->type == GGML_TYPE_F32;