Просмотр исходного кода

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

Jeff Bolz 7 месяцев назад
Родитель
Сommit
fef693dc6b
1 измененных файлов с 1 добавлено и 0 удалено
  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_RMS_NORM:
     case GGML_OP_CONV_2D_DW:
+    case GGML_OP_IM2COL:
         return true;
     default:
         return false;