whisper.rn 0.5.2 → 0.5.4

package/cpp/ggml-metal/ggml-metal-ops.h

@@ -52,6 +52,7 @@ int wsp_ggml_metal_op_unary             (wsp_ggml_metal_op_t ctx, int idx);
 int wsp_ggml_metal_op_glu               (wsp_ggml_metal_op_t ctx, int idx);
 int wsp_ggml_metal_op_sum               (wsp_ggml_metal_op_t ctx, int idx);
 int wsp_ggml_metal_op_sum_rows          (wsp_ggml_metal_op_t ctx, int idx);
+int wsp_ggml_metal_op_cumsum            (wsp_ggml_metal_op_t ctx, int idx);
 int wsp_ggml_metal_op_get_rows          (wsp_ggml_metal_op_t ctx, int idx);
 int wsp_ggml_metal_op_set_rows          (wsp_ggml_metal_op_t ctx, int idx);
 int wsp_ggml_metal_op_soft_max          (wsp_ggml_metal_op_t ctx, int idx);
@@ -70,6 +71,7 @@ int wsp_ggml_metal_op_group_norm        (wsp_ggml_metal_op_t ctx, int idx);
 int wsp_ggml_metal_op_norm              (wsp_ggml_metal_op_t ctx, int idx);
 int wsp_ggml_metal_op_rope              (wsp_ggml_metal_op_t ctx, int idx);
 int wsp_ggml_metal_op_im2col            (wsp_ggml_metal_op_t ctx, int idx);
+int wsp_ggml_metal_op_conv_2d           (wsp_ggml_metal_op_t ctx, int idx);
 int wsp_ggml_metal_op_conv_transpose_1d (wsp_ggml_metal_op_t ctx, int idx);
 int wsp_ggml_metal_op_conv_transpose_2d (wsp_ggml_metal_op_t ctx, int idx);
 int wsp_ggml_metal_op_upscale           (wsp_ggml_metal_op_t ctx, int idx);

package/cpp/ggml-metal/ggml-metal.cpp

@@ -197,6 +197,11 @@ static size_t wsp_ggml_backend_metal_buffer_type_get_alloc_size(wsp_ggml_backend
                 res += wsp_ggml_metal_op_flash_attn_ext_extra_blk(tensor);
                 res += wsp_ggml_metal_op_flash_attn_ext_extra_tmp(tensor);
             } break;
+        case WSP_GGML_OP_CUMSUM:
+        case WSP_GGML_OP_ARGSORT:
+            {
+                res *= 2;
+            } break;
         default:
             break;
     }