]> git.djapps.eu Git - pkg/ggml/sources/llama.cpp/commitdiff
sycl : update support conditions (#9394)
authorAlberto Cabrera Pérez <redacted>
Wed, 11 Sep 2024 00:53:42 +0000 (01:53 +0100)
committerGitHub <redacted>
Wed, 11 Sep 2024 00:53:42 +0000 (08:53 +0800)
* 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>
ggml/src/ggml-sycl.cpp

index 4f03b01e7083a8f884c6c617aae8fd4fd9ecec48..e603503996f2ea400be7c0f4c12aeb3cc56c5e70 100644 (file)
@@ -5137,13 +5137,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: