Skip to content

Commit b48a8c5

Browse files
committed
Fix logic of disabling exponent kernel
1 parent a65866b commit b48a8c5

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

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

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -4215,9 +4215,6 @@ static bool ggml_backend_sycl_device_supports_op(ggml_backend_dev_t dev, const g
42154215
case GGML_UNARY_OP_GELU_QUICK:
42164216
case GGML_UNARY_OP_GELU_ERF:
42174217
case GGML_UNARY_OP_TANH:
4218-
case GGML_UNARY_OP_EXP:
4219-
// Disable FP16 until we find out the root cause of failing fp16 sycl::exp
4220-
return ggml_is_contiguous(op->src[0]) && (op->type == op->src[0]->type) && op->src[0]->type == GGML_TYPE_F32;
42214218
case GGML_UNARY_OP_SGN:
42224219
case GGML_UNARY_OP_ABS:
42234220
case GGML_UNARY_OP_ELU:
@@ -4226,6 +4223,9 @@ static bool ggml_backend_sycl_device_supports_op(ggml_backend_dev_t dev, const g
42264223
#else
42274224
return ggml_is_contiguous(op->src[0]) && (op->src[0]->type == GGML_TYPE_F32 && op->type == GGML_TYPE_F32) && (op->type == op->src[0]->type);
42284225
#endif
4226+
case GGML_UNARY_OP_EXP:
4227+
// Disable FP16 until we find out the root cause of failing fp16 sycl::exp
4228+
return ggml_is_contiguous(op->src[0]) && (op->type == op->src[0]->type) && op->src[0]->type == GGML_TYPE_F32;
42294229
default:
42304230
return false;
42314231
}

0 commit comments

Comments
 (0)