mpi : fix after master merge

This commit is contained in:
Georgi Gerganov 2023-07-09 22:23:04 +03:00
parent 81c5ddd532
commit 0492363137
No known key found for this signature in database
GPG key ID: 449E073F9DC10735
3 changed files with 6 additions and 4 deletions

View file

@ -28,7 +28,8 @@ void ggml_mpi_graph_compute(
struct ggml_mpi_context * ctx_mpi,
struct ggml_context * ctx,
struct ggml_cgraph * gf,
int n_layers);
int n_layers,
int n_threads);
#ifdef __cplusplus
}