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

only use ggml_allocr_alloc when tensor has NULL data and is no view

Browse source
This commit is contained in:
xaedes 2023-08-14 20:56:56 +02:00
parent faf3e21eaf
commit 098654c277
No known key found for this signature in database
GPG key ID: 30030EDD817EA2B1
1 changed files with 6 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -986,11 +986,15 @@ struct ggml_tensor * llama_build_train_graphs(
// gradient tensors (will be set to zero by ggml_graph_reset)
for (int i = 0; i < gf->n_nodes; ++i) {
if (!gf->grads[i]) continue;
ggml_allocr_alloc(alloc, gf->grads[i]);
if (gf->grads[i]->data == NULL && !ggml_is_view(gf->grads[i])) {
ggml_allocr_alloc(alloc, gf->grads[i]);
}
ggml_build_forward_expand(gb, ggml_scale_inplace(ctx, gf->grads[i], one));
}
for (int i = 0; i < checkpoints.size(); ++i) {
ggml_allocr_alloc(alloc, checkpoints[i]);
if (checkpoints[i]->data == NULL && !ggml_is_view(checkpoints[i])) {
ggml_allocr_alloc(alloc, checkpoints[i]);
}
}
int n_leafs_after = gb->n_leafs;