ggerganov commited on
Commit
ed9150a
·
1 Parent(s): a13c99b

metal : update support condition for im2col + fix warning (llama/0)

Browse files
Files changed (1) hide show
  1. 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;