From 6216273edefbdd393ad8f986d4d86613e9886c7f Mon Sep 17 00:00:00 2001 From: Yee Man Chan Date: Thu, 29 Jan 2026 08:46:33 +0800 Subject: [PATCH] removed all ggml_cont b4 ggml_reshape_4d --- src/models/kimi-linear.cpp | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/models/kimi-linear.cpp b/src/models/kimi-linear.cpp index 5f497722d0..a7e5482008 100644 --- a/src/models/kimi-linear.cpp +++ b/src/models/kimi-linear.cpp @@ -540,9 +540,9 @@ std::pair llm_build_kimi_linear::build_kda_chunkin // decay_mask [S_k,BT_j,BT_i,CHB] *Note* second and third chunk_sizes are switched decay_mask = ggml_cont_4d(ctx0, ggml_permute(ctx0, decay_mask, 2, 1, 0, 3), S_k, chunk_size, chunk_size, CHB); - ggml_tensor * k_i = ggml_cont(ctx0, ggml_reshape_4d(ctx0, k, S_k, chunk_size, 1, CHB)); - ggml_tensor * k_j = ggml_cont(ctx0, ggml_reshape_4d(ctx0, k, S_k, 1, chunk_size, CHB)); - ggml_tensor * q_i = ggml_cont(ctx0, ggml_reshape_4d(ctx0, q, S_k, chunk_size, 1, CHB)); + ggml_tensor * k_i = ggml_reshape_4d(ctx0, k, S_k, chunk_size, 1, CHB); + ggml_tensor * k_j = ggml_reshape_4d(ctx0, k, S_k, 1, chunk_size, CHB); + ggml_tensor * q_i = ggml_reshape_4d(ctx0, q, S_k, chunk_size, 1, CHB); ggml_tensor * decay_k_i = ggml_mul(ctx0, decay_mask, k_i); ggml_tensor * decay_q_i = ggml_mul(ctx0, decay_mask, q_i);