Spaces:
Running
Running
metal : update support condition for im2col + fix warning (llama/0)
Browse files- ggml/src/ggml-metal.m +2 -1
ggml/src/ggml-metal.m
CHANGED
|
@@ -799,8 +799,9 @@ static bool ggml_metal_supports_op(const struct ggml_backend_metal_context * ctx
|
|
| 799 |
return ctx->support_simdgroup_reduction;
|
| 800 |
case GGML_OP_NORM:
|
| 801 |
case GGML_OP_ROPE:
|
| 802 |
-
case GGML_OP_IM2COL:
|
| 803 |
return true;
|
|
|
|
|
|
|
| 804 |
case GGML_OP_POOL_1D:
|
| 805 |
case GGML_OP_POOL_2D:
|
| 806 |
return false;
|
|
|
|
| 799 |
return ctx->support_simdgroup_reduction;
|
| 800 |
case GGML_OP_NORM:
|
| 801 |
case GGML_OP_ROPE:
|
|
|
|
| 802 |
return true;
|
| 803 |
+
case GGML_OP_IM2COL:
|
| 804 |
+
return op->src[0]->type == GGML_TYPE_F16;
|
| 805 |
case GGML_OP_POOL_1D:
|
| 806 |
case GGML_OP_POOL_2D:
|
| 807 |
return false;
|