Parcourir la source

sycl : update support conditions (#9394)

* sycl : update support condition to im2col

Signed-off-by: Alberto Cabrera <alberto.cabrera@codeplay.com>

* Added TODO to remind supporting FP32 im2col

---------

Signed-off-by: Alberto Cabrera <alberto.cabrera@codeplay.com>
Alberto Cabrera Pérez il y a 1 an
Parent
commit
51b6038636
1 fichiers modifiés avec 4 ajouts et 0 suppressions
  1. 4 0
      ggml/src/ggml-sycl.cpp

+ 4 - 0
ggml/src/ggml-sycl.cpp

@@ -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_SCALE:
         case GGML_OP_SQR:
         case GGML_OP_SQR:
         case GGML_OP_CLAMP:
         case GGML_OP_CLAMP:
+            return true;
         case GGML_OP_CONT:
         case GGML_OP_CONT:
+            return op->src[0]->type != GGML_TYPE_BF16;
         case GGML_OP_DIAG_MASK_INF:
         case GGML_OP_DIAG_MASK_INF:
         case GGML_OP_SOFT_MAX:
         case GGML_OP_SOFT_MAX:
             return true;
             return true;
         case GGML_OP_ROPE:
         case GGML_OP_ROPE:
             return ggml_is_contiguous(op->src[0]);
             return ggml_is_contiguous(op->src[0]);
         case GGML_OP_IM2COL:
         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_POOL_2D:
         case GGML_OP_SUM_ROWS:
         case GGML_OP_SUM_ROWS:
         case GGML_OP_ARGSORT:
         case GGML_OP_ARGSORT: