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 @@ -1846,7 +1846,7 @@ static bool ggml_metal_supports_op(const struct ggml_backend_metal_device_contex
1846
1846
case GGML_OP_ROPE:
1847
1847
return true ;
1848
1848
case GGML_OP_IM2COL:
1849
- return op->src [0 ]->type == GGML_TYPE_F16;
1849
+ return op->src [1 ]->type == GGML_TYPE_F32 && (op-> type == GGML_TYPE_F16 || op-> type == GGML_TYPE_F32) ;
1850
1850
case GGML_OP_POOL_1D:
1851
1851
return false ;
1852
1852
case GGML_OP_UPSCALE:
@@ -4703,7 +4703,6 @@ static int ggml_metal_encode_node(
4703
4703
{
4704
4704
GGML_ASSERT (ggml_is_contiguous (src0));
4705
4705
GGML_ASSERT (ggml_is_contiguous (src1));
4706
- GGML_ASSERT (src0->type == GGML_TYPE_F16);
4707
4706
GGML_ASSERT (src1->type == GGML_TYPE_F32);
4708
4707
GGML_ASSERT ( dst->type == GGML_TYPE_F16 || dst->type == GGML_TYPE_F32);
4709
4708
You can’t perform that action at this time.
0 commit comments