diff options
author | Spencer Sutton <spencersutton@users.noreply.github.com> | 2023-07-11 12:31:10 -0400 |
---|---|---|
committer | GitHub <noreply@github.com> | 2023-07-11 19:31:10 +0300 |
commit | 5bf2a2771886ee86137e01dbc7492f78fb392066 (patch) | |
tree | 740fd968318c00cfc4c001365f42d4e2cdc7f760 /ggml-mpi.c | |
parent | c9c74b4e3f9dcfab8b0032749ff8a579ab4e4d8d (diff) |
ggml : remove src0 and src1 from ggml_tensor and rename opt to src (#2178)
* Add ggml changes
* Update train-text-from-scratch for change
* mpi : adapt to new ggml_tensor->src
---------
Co-authored-by: Georgi Gerganov <ggerganov@gmail.com>
Diffstat (limited to 'ggml-mpi.c')
-rw-r--r-- | ggml-mpi.c | 8 |
1 files changed, 4 insertions, 4 deletions
@@ -175,11 +175,11 @@ void ggml_mpi_graph_compute_pre( // attach the input data to all nodes that need it // TODO: not great - should be able to do this without modifying the compute graph (see next TODO below) for (int i = idx_l0; i < idx_l1; i++) { - if (gf->nodes[i]->src0 == gf->nodes[idx_l0]) { - gf->nodes[i]->src0 = inp0; + if (gf->nodes[i]->src[0] == gf->nodes[idx_l0]) { + gf->nodes[i]->src[0] = inp0; } - if (gf->nodes[i]->src1 == gf->nodes[idx_l0]) { - gf->nodes[i]->src1 = inp0; + if (gf->nodes[i]->src[1] == gf->nodes[idx_l0]) { + gf->nodes[i]->src[1] = inp0; } } |