Skip to content

Commit

Permalink
sycl : update support conditions (#9394)
Browse files Browse the repository at this point in the history
* 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]>
  • Loading branch information
Alcpz authored Sep 11, 2024
1 parent cb9c933 commit 51b6038
Showing 1 changed file with 4 additions and 0 deletions.
4 changes: 4 additions & 0 deletions ggml/src/ggml-sycl.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -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:
Expand Down

0 comments on commit 51b6038

Please sign in to comment.