batched-bench : fix oob write

ggml-ci
This commit is contained in:
Georgi Gerganov
2025-07-04 13:08:31 +03:00
parent b123d89445
commit ab8443d0ea
2 changed files with 6 additions and 2 deletions

View File

@ -224,6 +224,7 @@ int main(int argc, char ** argv) {
auto & client = clients[i]; auto & client = clients[i];
client.id = i; client.id = i;
client.smpl = common_sampler_init(model, params.sampling); client.smpl = common_sampler_init(model, params.sampling);
//params.sampling.seed++;
} }
std::vector<llama_token> tokens_system; std::vector<llama_token> tokens_system;
@ -345,7 +346,7 @@ int main(int argc, char ** argv) {
client.n_decoded = 0; client.n_decoded = 0;
client.i_batch = batch.n_tokens - 1; client.i_batch = batch.n_tokens - 1;
LOG_INF("\033[31mClient %3d, seq %4d, junk = %4d, started decoding ...\033[0m\n", client.id, client.seq_id, n_junk_cur); LOG_INF("\033[31mClient %3d, seq %4d, junk = %4d, prompt = %d, started decoding ...\033[0m\n", client.id, client.seq_id, n_junk_cur, client.n_prompt);
g_seq_id += 1; g_seq_id += 1;

View File

@ -130,7 +130,10 @@ int main(int argc, char ** argv) {
common_batch_add(batch, 0, i, { j }, false); common_batch_add(batch, 0, i, { j }, false);
} }
} }
batch.logits[batch.n_tokens - 1] = true;
if (batch.n_tokens > 0) {
batch.logits[batch.n_tokens - 1] = true;
}
const auto t_pp_start = ggml_time_us(); const auto t_pp_start = ggml_time_us();