Skip to content

Commit 51b6038

Browse files
authored
sycl : update support conditions (ggml-org#9394)
* sycl : update support condition to im2col Signed-off-by: Alberto Cabrera <[email protected]> * Added TODO to remind supporting FP32 im2col --------- Signed-off-by: Alberto Cabrera <[email protected]>
1 parent cb9c933 commit 51b6038

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

ggml/src/ggml-sycl.cpp

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -5137,13 +5137,17 @@ GGML_CALL static bool ggml_backend_sycl_supports_op(ggml_backend_t backend, cons
51375137
case GGML_OP_SCALE:
51385138
case GGML_OP_SQR:
51395139
case GGML_OP_CLAMP:
5140+
return true;
51405141
case GGML_OP_CONT:
5142+
return op->src[0]->type != GGML_TYPE_BF16;
51415143
case GGML_OP_DIAG_MASK_INF:
51425144
case GGML_OP_SOFT_MAX:
51435145
return true;
51445146
case GGML_OP_ROPE:
51455147
return ggml_is_contiguous(op->src[0]);
51465148
case GGML_OP_IM2COL:
5149+
// TODO: add support for the new F32 operations
5150+
return op->src[0]->type == GGML_TYPE_F16;
51475151
case GGML_OP_POOL_2D:
51485152
case GGML_OP_SUM_ROWS:
51495153
case GGML_OP_ARGSORT:

0 commit comments

Comments
 (0)