From 5a11b75875e2e82501c1bdaf7c4528a3ecc4f4e3 Mon Sep 17 00:00:00 2001 From: xaedes Date: Mon, 14 Aug 2023 17:55:51 +0200 Subject: [PATCH] fix variable names --- examples/train-text-from-scratch/train-text-from-scratch.cpp | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/examples/train-text-from-scratch/train-text-from-scratch.cpp b/examples/train-text-from-scratch/train-text-from-scratch.cpp index 410ba69b9..bdc7cdade 100644 --- a/examples/train-text-from-scratch/train-text-from-scratch.cpp +++ b/examples/train-text-from-scratch/train-text-from-scratch.cpp @@ -1517,9 +1517,9 @@ void ggml_build_backward_gradient_checkpointing( // insert checkpoints in replacements for (int i = 0; i < n_checkpoints; ++i) { - size_t k = hash_find(replacements->keys, node); + size_t k = hash_find(replacements->keys, checkpoints[i]); GGML_ASSERT(k < GGML_GRAPH_HASHTABLE_SIZE); // assert that not full - GGML_ASSERT(replacements->keys[i] == NULL); // assert that we don't overwrite + GGML_ASSERT(replacements->keys[k] == NULL); // assert that we don't overwrite replacements->keys[k] = checkpoints[i]; replacements->vals[k] = checkpoints[i]; }