metal : add cumsum (#17305)

2025-11-17 11:51:13 +02:00
parent 2376b7758c
commit 1a139644a8
9 changed files with 406 additions and 79 deletions
@@ -197,6 +197,7 @@ static size_t ggml_backend_metal_buffer_type_get_alloc_size(ggml_backend_buffer_
                res += ggml_metal_op_flash_attn_ext_extra_blk(tensor);
                res += ggml_metal_op_flash_attn_ext_extra_tmp(tensor);
            } break;
+        case GGML_OP_CUMSUM:
        case GGML_OP_ARGSORT:
            {
                res *= 2;