This commit is contained in:
Georgi Gerganov 2024-05-20 17:00:55 +03:00
parent 26cd4237bc
commit a041ced0fd
No known key found for this signature in database
GPG key ID: BF970631944C16B7
4 changed files with 176 additions and 96 deletions

View file

@ -156,7 +156,15 @@ int main(int argc, char ** argv) {
const auto t_main_start = ggml_time_us();
// debug
struct llama_kv_cache_view kvc_view = llama_kv_cache_view_init(ctx, 1);
while (n_cur <= n_len) {
if (false) {
llama_kv_cache_view_update(ctx, &kvc_view);
dump_kv_cache_view_seqs(kvc_view, 40);
}
// prepare the next batch
llama_batch_clear(batch);