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
18461846        case  GGML_OP_ROPE:
18471847            return  true ;
18481848        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) ;
18501850        case  GGML_OP_POOL_1D:
18511851            return  false ;
18521852        case  GGML_OP_UPSCALE:
@@ -4703,7 +4703,6 @@ static int ggml_metal_encode_node(
47034703            {
47044704                GGML_ASSERT (ggml_is_contiguous (src0));
47054705                GGML_ASSERT (ggml_is_contiguous (src1));
4706-                 GGML_ASSERT (src0->type  == GGML_TYPE_F16);
47074706                GGML_ASSERT (src1->type  == GGML_TYPE_F32);
47084707                GGML_ASSERT ( dst->type  == GGML_TYPE_F16 || dst->type  == GGML_TYPE_F32);
47094708
    
 
   
 
     
   
   
          
     
  
    
     
 
    
      
     
 
     
    You can’t perform that action at this time.
  
 
    
  
     
    
      
        
     
 
       
      
     
   
 
    
    
  
 
  
 
     
    
0 commit comments