From: Alberto Cabrera Pérez <redacted>
Date: Wed, 11 Sep 2024 00:53:42 +0000 (+0100)
Subject: sycl : update support conditions (llama/9394)
X-Git-Tag: upstream/0.0.1642~377
X-Git-Url: https://git.djapps.eu/?a=commitdiff_plain;h=077a9bc973cd03d12409179e4fad61204a6ade1d;p=pkg%2Fggml%2Fsources%2Fggml

sycl : update support conditions (llama/9394)

* sycl : update support condition to im2col

Signed-off-by: Alberto Cabrera <redacted>

* Added TODO to remind supporting FP32 im2col

---------

Signed-off-by: Alberto Cabrera <redacted>
---

diff --git a/src/ggml-sycl.cpp b/src/ggml-sycl.cpp
index 4106244d..fab97161 100644
--- a/src/ggml-sycl.cpp
+++ b/src/ggml-sycl.cpp
@@ -5138,13 +5138,17 @@ GGML_CALL static bool ggml_backend_sycl_supports_op(ggml_backend_t backend, cons
         case GGML_OP_SCALE:
         case GGML_OP_SQR:
         case GGML_OP_CLAMP:
+            return true;
         case GGML_OP_CONT:
+            return op->src[0]->type != GGML_TYPE_BF16;
         case GGML_OP_DIAG_MASK_INF:
         case GGML_OP_SOFT_MAX:
             return true;
         case GGML_OP_ROPE:
             return ggml_is_contiguous(op->src[0]);
         case GGML_OP_IM2COL:
+            // TODO: add support for the new F32 operations
+            return op->src[0]->type == GGML_TYPE_F16;
         case GGML_OP_POOL_2D:
         case GGML_OP_SUM_ROWS:
         case GGML_OP_ARGSORT: