Skip to content

Commit 7823459

Browse files
Alcpzarthw
authored andcommitted
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 51d48d0 commit 7823459

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
@@ -5014,13 +5014,17 @@ GGML_CALL static bool ggml_backend_sycl_supports_op(ggml_backend_t backend, cons
50145014
case GGML_OP_SCALE:
50155015
case GGML_OP_SQR:
50165016
case GGML_OP_CLAMP:
5017+
return true;
50175018
case GGML_OP_CONT:
5019+
return op->src[0]->type != GGML_TYPE_BF16;
50185020
case GGML_OP_DIAG_MASK_INF:
50195021
case GGML_OP_SOFT_MAX:
50205022
return true;
50215023
case GGML_OP_ROPE:
50225024
return ggml_is_contiguous(op->src[0]);
50235025
case GGML_OP_IM2COL:
5026+
// TODO: add support for the new F32 operations
5027+
return op->src[0]->type == GGML_TYPE_F16;
50245028
case GGML_OP_POOL_2D:
50255029
case GGML_OP_SUM_ROWS:
50265030
case GGML_OP_ARGSORT:

0 commit comments

Comments
 (0)