llama_context: synchronize before reallocating output buffer (#17974)

diff --git a/src/llama-context.cpp b/src/llama-context.cpp
index 2692297dcaba20116e49a601d4135c238ce4f596..9914b3276b7092f61b467f5346c2abc77ecc7c1a 100644 (file)
--- a/src/llama-context.cpp
+++ b/src/llama-context.cpp
@@ -1318,6 +1318,7 @@ uint32_t llama_context::output_reserve(int32_t n_outputs) {
             // This doesn't happen often, but may be annoying in some cases (like the HellaSwag benchmark)
             LLAMA_LOG_INFO("%s: reallocating output buffer from size %.02f MiB to %.02f MiB\n", __func__, prev_size / 1024.0 / 1024.0, new_size / 1024.0 / 1024.0);
 #endif
+            synchronize();
             buf_output = nullptr;
             logits = nullptr;
             embd = nullptr;