aboutsummaryrefslogtreecommitdiff
path: root/ggml-mpi.c
diff options
context:
space:
mode:
authorSpencer Sutton <spencersutton@users.noreply.github.com>2023-07-11 12:31:10 -0400
committerGitHub <noreply@github.com>2023-07-11 19:31:10 +0300
commit5bf2a2771886ee86137e01dbc7492f78fb392066 (patch)
tree740fd968318c00cfc4c001365f42d4e2cdc7f760 /ggml-mpi.c
parentc9c74b4e3f9dcfab8b0032749ff8a579ab4e4d8d (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.c8
1 files changed, 4 insertions, 4 deletions
diff --git a/ggml-mpi.c b/ggml-mpi.c
index 872e808..ae176d7 100644
--- a/ggml-mpi.c
+++ b/ggml-mpi.c
@@ -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;
}
}