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

mpi send use input as output (#1750)

Browse Source 
* mpi send use input as output

* move earlier
This commit is contained in:
Awni Hannun
2025-01-06 06:08:43 -08:00
committed by GitHub
parent eab93985b8
commit 058d6ce683
5 changed files with 19 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
python/src/distributed.cpp
View File

@@ -148,7 +148,7 @@ void init_distributed(nb::module_& parent_module) {
in which case the default stream of the default device is used.
Returns:
array: An empty array which when evaluated the send is performed.
array: An array identical to ``x`` which when evaluated the send is performed.
)pbdoc");
m.def(