Skip to content

Commit 9188856

Browse files
committed
llama : resolve rwkv conflict
ggml-ci
1 parent e665b57 commit 9188856

File tree

1 file changed

+3
-9
lines changed

1 file changed

+3
-9
lines changed

src/llama.cpp

Lines changed: 3 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -7076,19 +7076,13 @@ struct llm_build_context {
70767076
// 1
70777077
// );
70787078

7079+
// struct ggml_tensor * last_norm_att = ggml_view_3d(ctx0, x_norm_att, n_embd, 1, n_seqs, x_norm_att->nb[1], x_norm_att->nb[2], (n_seq_tokens-1)*n_embd*ggml_element_size(x_norm_att));
70797080
// ggml_build_forward_expand(
70807081
// gf,
70817082
// ggml_cpy(
70827083
// ctx0,
7083-
// wkv_states,
7084-
// ggml_view_1d(
7085-
// ctx0,
7086-
// kv_self.v_l[il],
7087-
// hparams.n_embd_v_s() * n_seqs,
7088-
// hparams.n_embd_v_s() * kv_head * ggml_element_size(kv_self.v_l[il])
7089-
// )
7090-
// )
7091-
// );
7084+
// ggml_view_1d(ctx0, last_norm_att, n_embd * n_seqs, 0),
7085+
// ggml_view_1d(ctx0, kv_self.k_l[il], hparams.n_embd_k_s() * n_seqs, hparams.n_embd_k_s() * kv_head * ggml_element_size(kv_self.k_l[il]))
70927086

70937087
// struct ggml_tensor * ffn_inp = ggml_add(ctx0, cur, build_rwkv6_time_mix(layer, x_norm_att, x_prev, &wkv_states, hparams.wkv_head_size, hparams.n_head_kv()));
70947088
// ggml_build_forward_expand(gf, ffn_inp);

0 commit comments

Comments
 (0)