mpi : add names for layer inputs + prep ggml_mpi_graph_compute()

2023-07-09 14:42:36 +03:00 · 2023-07-09 14:42:36 +03:00 · e339d35579
commit e339d35579
parent 3232db628c
3 changed files with 46 additions and 49 deletions
--- a/ggml-mpi.c
+++ b/ggml-mpi.c
@ -135,3 +135,12 @@ struct ggml_tensor * ggml_mpi_eval_init(

    return res;
 }
+
+void ggml_mpi_graph_compute(
+        struct ggml_mpi_context * ctx_mpi,
+             struct ggml_cgraph * gf,
+                            int   n_layers,
+                            int   n_embd,
+                            int   n_tokens) {
+
+}