File tree Expand file tree Collapse file tree 1 file changed +1
-2
lines changed Expand file tree Collapse file tree 1 file changed +1
-2
lines changed Original file line number Diff line number Diff line change @@ -1876,7 +1876,7 @@ static bool ggml_metal_supports_op(const struct ggml_backend_metal_device_contex
1876
1876
case GGML_OP_ROPE:
1877
1877
return true ;
1878
1878
case GGML_OP_IM2COL:
1879
- return op->src [1 ]->type == GGML_TYPE_F32 && (op->type == GGML_TYPE_F16 || op->type == GGML_TYPE_F32);
1879
+ return ggml_is_contiguous (op-> src [ 1 ]) && op->src [1 ]->type == GGML_TYPE_F32 && (op->type == GGML_TYPE_F16 || op->type == GGML_TYPE_F32);
1880
1880
case GGML_OP_POOL_1D:
1881
1881
return false ;
1882
1882
case GGML_OP_UPSCALE:
@@ -4731,7 +4731,6 @@ static int ggml_metal_encode_node(
4731
4731
} break ;
4732
4732
case GGML_OP_IM2COL:
4733
4733
{
4734
- GGML_ASSERT (ggml_is_contiguous (src0));
4735
4734
GGML_ASSERT (ggml_is_contiguous (src1));
4736
4735
GGML_ASSERT (src1->type == GGML_TYPE_F32);
4737
4736
GGML_ASSERT ( dst->type == GGML_TYPE_F16 || dst->type == GGML_TYPE_F32);
You can’t perform that action at this time.
0 commit comments