Skip to content

Commit a3dc326

Browse files
committed
update for bf16
1 parent f85657b commit a3dc326

1 file changed

Lines changed: 5 additions & 1 deletion

File tree

ggml/src/ggml-sycl/ggml-sycl.cpp

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5675,7 +5675,11 @@ static bool do_ggml_backend_sycl_device_supports_op(ggml_backend_dev_t dev, cons
56755675
return true;
56765676
case GGML_OP_COL2IM_1D:
56775677
return ggml_is_contiguous(op->src[0]) &&
5678-
(op->type == GGML_TYPE_F32 || op->type == GGML_TYPE_F16 || op->type == GGML_TYPE_BF16) &&
5678+
(op->type == GGML_TYPE_F32 || op->type == GGML_TYPE_F16
5679+
#ifdef GGML_SYCL_HAS_BF16
5680+
|| op->type == GGML_TYPE_BF16
5681+
#endif
5682+
) &&
56795683
op->src[0]->type == op->type;
56805684
case GGML_OP_CONV_3D:
56815685
return op->type == GGML_TYPE_F32 &&

0 commit comments

Comments
 (0)