Kaynağa Gözat

vulkan: mark IM2COL as supporting non-contig (#13783)

Jeff Bolz 7 ay önce
ebeveyn
işleme
fef693dc6b
1 değiştirilmiş dosya ile 1 ekleme ve 0 silme
  1. 1 0
      ggml/src/ggml-vulkan/ggml-vulkan.cpp

+ 1 - 0
ggml/src/ggml-vulkan/ggml-vulkan.cpp

@@ -6452,6 +6452,7 @@ static bool ggml_vk_op_supports_incontiguous(ggml_op op) {
     case GGML_OP_ROPE:
     case GGML_OP_ROPE:
     case GGML_OP_RMS_NORM:
     case GGML_OP_RMS_NORM:
     case GGML_OP_CONV_2D_DW:
     case GGML_OP_CONV_2D_DW:
+    case GGML_OP_IM2COL:
         return true;
         return true;
     default:
     default:
         return false;
         return false;