zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-12-16 01:49:05 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

WIP (distributed)

Browse Source
This commit is contained in:
Ronan Collobert
2025-10-31 13:32:56 -07:00
parent 19ab7911f6
commit c5913131cf
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
mlx/distributed/mpi/mpi.cpp
View File

@@ -36,7 +36,7 @@ void simple_sum(
void* input, void* input,
void* accumulator, void* accumulator,
int* len, int* len,
MPI_Datatype* datatype) { MPI_Datatype* /* datatype */) {
T* in = (T*)input; T* in = (T*)input;
T* acc = (T*)accumulator; T* acc = (T*)accumulator;
int N = *len; int N = *len;
@@ -55,7 +55,7 @@ void simple_max(
void* input, void* input,
void* accumulator, void* accumulator,
int* len, int* len,
MPI_Datatype* datatype) { MPI_Datatype* /* datatype */) {
T* in = (T*)input; T* in = (T*)input;
T* acc = (T*)accumulator; T* acc = (T*)accumulator;
int N = *len; int N = *len;
@@ -75,7 +75,7 @@ void simple_min(
void* input, void* input,
void* accumulator, void* accumulator,
int* len, int* len,
MPI_Datatype* datatype) { MPI_Datatype* /* datatype */) {
T* in = (T*)input; T* in = (T*)input;
T* acc = (T*)accumulator; T* acc = (T*)accumulator;
int N = *len; int N = *len;