Skip to content

Commit f706358

Browse files
ngxsonggerganov
andauthored
Update src/llama-kv-cache.cpp
Co-authored-by: Georgi Gerganov <[email protected]>
1 parent 9102a7c commit f706358

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

src/llama-kv-cache.cpp

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -901,9 +901,10 @@ void llama_kv_cache::apply_ubatch(const slot_info & sinfo, const llama_ubatch &
901901
cells.pos_set(idx, ubatch.pos[i]);
902902

903903
if (ubatch.is_pos_2d()) {
904-
llama_kv_cell_ext ext;
905-
ext.x = ubatch.pos[i + ubatch.n_tokens*2];
906-
ext.y = ubatch.pos[i + ubatch.n_tokens];
904+
llama_kv_cell_ext ext {
905+
/*.x =*/ ubatch.pos[i + ubatch.n_tokens*2],
906+
/*.y =*/ ubatch.pos[i + ubatch.n_tokens],
907+
};
907908
cells.ext_set(idx, std::move(ext));
908909
}
909910

0 commit comments

Comments
 (0)