vbatts/llama.cpp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

terminate recursive tensor cloning when reaching tensor without src tensors

Browse source
This commit is contained in:
xaedes 2023-08-14 17:53:36 +02:00
parent 0dd496c5e2
commit 52c92c0a8c
No known key found for this signature in database
GPG key ID: 30030EDD817EA2B1
1 changed files with 11 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

11
examples/train-text-from-scratch/train-text-from-scratch.cpp
View file

@ -1423,6 +1423,17 @@ struct ggml_tensor * ggml_recompute_graph_node(
return node; return node;
} }
int count_children = 0;
for (int k = 0; k < GGML_MAX_SRC; ++k) {
if (node->src[k]) {
++count_children;
}
}
if (count_children == 0) {
return node;
}
size_t i = hash_find(replacements->keys, node); size_t i = hash_find(replacements->keys, node);
GGML_ASSERT(i < GGML_GRAPH_HASHTABLE_SIZE); // assert that not full GGML_ASSERT(i < GGML_GRAPH_HASHTABLE_SIZE); // assert that not full
if (replacements->keys[i] == node) { if (replacements->keys[i] == node) {