|
@@ -735,6 +735,12 @@ static id<MTLBuffer> ggml_metal_get_buffer(struct ggml_tensor * t, size_t * offs
|
|
|
}
|
|
}
|
|
|
|
|
|
|
|
static bool ggml_metal_supports_op(const struct ggml_metal_context * ctx, const struct ggml_tensor * op) {
|
|
static bool ggml_metal_supports_op(const struct ggml_metal_context * ctx, const struct ggml_tensor * op) {
|
|
|
|
|
+ for (size_t i = 0, n = 3; i < n; ++i) {
|
|
|
|
|
+ if (op->src[i] != NULL && op->src[i]->type == GGML_TYPE_BF16) {
|
|
|
|
|
+ return false;
|
|
|
|
|
+ }
|
|
|
|
|
+ }
|
|
|
|
|
+
|
|
|
switch (op->op) {
|
|
switch (op->op) {
|
|
|
case GGML_OP_UNARY:
|
|
case GGML_OP_UNARY:
|
|
|
switch (ggml_get_unary_op(op)) {
|
|
switch (ggml_get_unary_op(op)) {
|