Public/llama.cpp
1
0
Fork 0
mirror of https://github.com/ggml-org/llama.cpp.git synced 2026-09-18 16:55:05 +02:00
Code Issues Packages Projects Releases Wiki Activity

common : add llama_batch_add() and llama_batch_clear() helpers

Browse Source
This commit is contained in:
Georgi Gerganov
2023-10-16 12:41:33 +03:00
parent 005949109d
commit 360a333145
10 changed files with 98 additions and 122 deletions
Download Patch File Download Diff File Expand all files Collapse all files
llama.cpp
+1 -1
View File
@@ -9342,7 +9342,7 @@ struct llama_batch llama_batch_get_one(
}
struct llama_batch llama_batch_init(int32_t n_tokens, int32_t embd, int32_t n_seq_max) {
llama_batch batch = { -1, nullptr, nullptr, nullptr, nullptr, nullptr, nullptr, 0, 0, 0, };
llama_batch batch = { 0, nullptr, nullptr, nullptr, nullptr, nullptr, nullptr, 0, 0, 0, };
if (embd) {
batch.embd = (float *) malloc(sizeof(float) * n_tokens * embd);