This commit is contained in:
Awni Hannun
2024-09-17 12:06:14 -07:00
committed by CircleCI Docs
parent 27e07888b0
commit 6a6ffb598d
739 changed files with 28107 additions and 8524 deletions

View File

@@ -83,6 +83,7 @@ $(function() {
<div class="contents">
<div class="textblock"><code>#include &lt;optional&gt;</code><br />
<code>#include &quot;<a class="el" href="distributed_8h_source.html">mlx/distributed/distributed.h</a>&quot;</code><br />
<code>#include &quot;<a class="el" href="utils_8h_source.html">mlx/utils.h</a>&quot;</code><br />
</div>
<p><a href="distributed_2ops_8h_source.html">Go to the source code of this file.</a></p>
<table class="memberdecls">
@@ -97,10 +98,16 @@ Namespaces</h2></td></tr>
</table><table class="memberdecls">
<tr class="heading"><td colspan="2"><h2 class="groupheader"><a id="func-members" name="func-members"></a>
Functions</h2></td></tr>
<tr class="memitem:ad1c6aa544781765060552cacfa845001" id="r_ad1c6aa544781765060552cacfa845001"><td class="memItemLeft" align="right" valign="top"><a class="el" href="classmlx_1_1core_1_1array.html">array</a>&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="namespacemlx_1_1core_1_1distributed.html#ad1c6aa544781765060552cacfa845001">mlx::core::distributed::all_sum</a> (const <a class="el" href="classmlx_1_1core_1_1array.html">array</a> &amp;x, std::optional&lt; <a class="el" href="structmlx_1_1core_1_1distributed_1_1_group.html">Group</a> &gt; group=std::nullopt)</td></tr>
<tr class="separator:ad1c6aa544781765060552cacfa845001"><td class="memSeparator" colspan="2">&#160;</td></tr>
<tr class="memitem:abf45499dd1209ba7162908f14634d2ba" id="r_abf45499dd1209ba7162908f14634d2ba"><td class="memItemLeft" align="right" valign="top"><a class="el" href="classmlx_1_1core_1_1array.html">array</a>&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="namespacemlx_1_1core_1_1distributed.html#abf45499dd1209ba7162908f14634d2ba">mlx::core::distributed::all_gather</a> (const <a class="el" href="classmlx_1_1core_1_1array.html">array</a> &amp;x, std::optional&lt; <a class="el" href="structmlx_1_1core_1_1distributed_1_1_group.html">Group</a> &gt; group=std::nullopt)</td></tr>
<tr class="separator:abf45499dd1209ba7162908f14634d2ba"><td class="memSeparator" colspan="2">&#160;</td></tr>
<tr class="memitem:a67ccb1a5445fc6f5db49dd36a15e5980" id="r_a67ccb1a5445fc6f5db49dd36a15e5980"><td class="memItemLeft" align="right" valign="top"><a class="el" href="classmlx_1_1core_1_1array.html">array</a>&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="namespacemlx_1_1core_1_1distributed.html#a67ccb1a5445fc6f5db49dd36a15e5980">mlx::core::distributed::all_sum</a> (const <a class="el" href="classmlx_1_1core_1_1array.html">array</a> &amp;x, std::optional&lt; <a class="el" href="structmlx_1_1core_1_1distributed_1_1_group.html">Group</a> &gt; group=std::nullopt, <a class="el" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s={})</td></tr>
<tr class="separator:a67ccb1a5445fc6f5db49dd36a15e5980"><td class="memSeparator" colspan="2">&#160;</td></tr>
<tr class="memitem:a82ef5e8cc7ac62cd228e51b1c1b77cb7" id="r_a82ef5e8cc7ac62cd228e51b1c1b77cb7"><td class="memItemLeft" align="right" valign="top"><a class="el" href="classmlx_1_1core_1_1array.html">array</a>&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="namespacemlx_1_1core_1_1distributed.html#a82ef5e8cc7ac62cd228e51b1c1b77cb7">mlx::core::distributed::all_gather</a> (const <a class="el" href="classmlx_1_1core_1_1array.html">array</a> &amp;x, std::optional&lt; <a class="el" href="structmlx_1_1core_1_1distributed_1_1_group.html">Group</a> &gt; group=std::nullopt, <a class="el" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> S={})</td></tr>
<tr class="separator:a82ef5e8cc7ac62cd228e51b1c1b77cb7"><td class="memSeparator" colspan="2">&#160;</td></tr>
<tr class="memitem:a5a8360edaa3a528a3927fce4d2cf1777" id="r_a5a8360edaa3a528a3927fce4d2cf1777"><td class="memItemLeft" align="right" valign="top"><a class="el" href="classmlx_1_1core_1_1array.html">array</a>&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="namespacemlx_1_1core_1_1distributed.html#a5a8360edaa3a528a3927fce4d2cf1777">mlx::core::distributed::send</a> (const <a class="el" href="classmlx_1_1core_1_1array.html">array</a> &amp;x, int dst, std::optional&lt; <a class="el" href="structmlx_1_1core_1_1distributed_1_1_group.html">Group</a> &gt; group=std::nullopt, <a class="el" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s={})</td></tr>
<tr class="separator:a5a8360edaa3a528a3927fce4d2cf1777"><td class="memSeparator" colspan="2">&#160;</td></tr>
<tr class="memitem:a10f9d39c02e6e5db600912c03de8b393" id="r_a10f9d39c02e6e5db600912c03de8b393"><td class="memItemLeft" align="right" valign="top"><a class="el" href="classmlx_1_1core_1_1array.html">array</a>&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="namespacemlx_1_1core_1_1distributed.html#a10f9d39c02e6e5db600912c03de8b393">mlx::core::distributed::recv</a> (std::vector&lt; int &gt; shape, <a class="el" href="structmlx_1_1core_1_1_dtype.html">Dtype</a> dtype, int src, std::optional&lt; <a class="el" href="structmlx_1_1core_1_1distributed_1_1_group.html">Group</a> &gt; group=std::nullopt, <a class="el" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s={})</td></tr>
<tr class="separator:a10f9d39c02e6e5db600912c03de8b393"><td class="memSeparator" colspan="2">&#160;</td></tr>
<tr class="memitem:a2822b78bce2c679e6ff940b2fca944f0" id="r_a2822b78bce2c679e6ff940b2fca944f0"><td class="memItemLeft" align="right" valign="top"><a class="el" href="classmlx_1_1core_1_1array.html">array</a>&#160;</td><td class="memItemRight" valign="bottom"><a class="el" href="namespacemlx_1_1core_1_1distributed.html#a2822b78bce2c679e6ff940b2fca944f0">mlx::core::distributed::recv_like</a> (const <a class="el" href="classmlx_1_1core_1_1array.html">array</a> &amp;x, int src, std::optional&lt; <a class="el" href="structmlx_1_1core_1_1distributed_1_1_group.html">Group</a> &gt; group=std::nullopt, <a class="el" href="namespacemlx_1_1core.html#a95fc1013cc48fbfee0c54310711a5e58">StreamOrDevice</a> s={})</td></tr>
<tr class="separator:a2822b78bce2c679e6ff940b2fca944f0"><td class="memSeparator" colspan="2">&#160;</td></tr>
</table>
</div><!-- contents -->
<!-- start footer part -->