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

rebase

Browse Source
This commit is contained in:
CircleCI Docs
2025-03-20 22:37:22 +00:00
parent a943912d4c
commit cecec56a99
858 changed files with 18494 additions and 17475 deletions
Download Patch File Download Diff File Expand all files Collapse all files

46
docs/build/html/distributed__impl_8h_source.html vendored
View File

@@ -122,44 +122,40 @@ $(function(){initNavTree('distributed__impl_8h_source.html',''); initResizable(t
<div class="line"><a id="l00017" name="l00017"></a><span class="lineno"><a class="line" href="classmlx_1_1core_1_1distributed_1_1detail_1_1_group_impl.html#ab1c8044b05f185c4bcc53002d4587599"> 17</a></span> <span class="keyword">virtual</span> <span class="keywordtype">int</span> <a class="code hl_function" href="classmlx_1_1core_1_1distributed_1_1detail_1_1_group_impl.html#ab1c8044b05f185c4bcc53002d4587599">size</a>() = 0;</div>
<div class="line"><a id="l00018" name="l00018"></a><span class="lineno"><a class="line" href="classmlx_1_1core_1_1distributed_1_1detail_1_1_group_impl.html#a87800a23c8160933a2d77a55a959194d"> 18</a></span> <span class="keyword">virtual</span> std::shared_ptr&lt;GroupImpl&gt; <a class="code hl_function" href="classmlx_1_1core_1_1distributed_1_1detail_1_1_group_impl.html#a87800a23c8160933a2d77a55a959194d">split</a>(<span class="keywordtype">int</span> color, <span class="keywordtype">int</span> key = -1) = 0;</div>
<div class="line"><a id="l00019" name="l00019"></a><span class="lineno"> 19</span> </div>
<div class="line"><a id="l00020" name="l00020"></a><span class="lineno"><a class="line" href="classmlx_1_1core_1_1distributed_1_1detail_1_1_group_impl.html#ae163a6f444c6cc8820288b20f294e483"> 20</a></span> <span class="keyword">virtual</span> <span class="keywordtype">void</span> <a class="code hl_function" href="classmlx_1_1core_1_1distributed_1_1detail_1_1_group_impl.html#ae163a6f444c6cc8820288b20f294e483">all_sum</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>&amp; input, <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>&amp; output) = 0;</div>
<div class="line"><a id="l00021" name="l00021"></a><span class="lineno"><a class="line" href="classmlx_1_1core_1_1distributed_1_1detail_1_1_group_impl.html#a04bb1df23abe5b1f3fa0126375c6cea4"> 21</a></span> <span class="keyword">virtual</span> <span class="keywordtype">void</span> <a class="code hl_function" href="classmlx_1_1core_1_1distributed_1_1detail_1_1_group_impl.html#a04bb1df23abe5b1f3fa0126375c6cea4">all_gather</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>&amp; input, <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>&amp; output) = 0;</div>
<div class="line"><a id="l00022" name="l00022"></a><span class="lineno"><a class="line" href="classmlx_1_1core_1_1distributed_1_1detail_1_1_group_impl.html#ac8472eb2f96d1b14c7e4ccef56268ba0"> 22</a></span> <span class="keyword">virtual</span> <span class="keywordtype">void</span> <a class="code hl_function" href="classmlx_1_1core_1_1distributed_1_1detail_1_1_group_impl.html#ac8472eb2f96d1b14c7e4ccef56268ba0">send</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>&amp; input, <span class="keywordtype">int</span> dst) = 0;</div>
<div class="line"><a id="l00023" name="l00023"></a><span class="lineno"><a class="line" href="classmlx_1_1core_1_1distributed_1_1detail_1_1_group_impl.html#ac4af5fc16a82ba8c72df04d7694f8352"> 23</a></span> <span class="keyword">virtual</span> <span class="keywordtype">void</span> <a class="code hl_function" href="classmlx_1_1core_1_1distributed_1_1detail_1_1_group_impl.html#ac4af5fc16a82ba8c72df04d7694f8352">recv</a>(<a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>&amp; out, <span class="keywordtype">int</span> src) = 0;</div>
<div class="line"><a id="l00020" name="l00020"></a><span class="lineno"><a class="line" href="classmlx_1_1core_1_1distributed_1_1detail_1_1_group_impl.html#a23f620ec55d75f236d5371e05a52fd64"> 20</a></span> <span class="keyword">virtual</span> <span class="keywordtype">void</span> <a class="code hl_function" href="classmlx_1_1core_1_1distributed_1_1detail_1_1_group_impl.html#a23f620ec55d75f236d5371e05a52fd64">all_sum</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>&amp; input, <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>&amp; output, <a class="code hl_struct" href="structmlx_1_1core_1_1_stream.html">Stream</a> stream) = 0;</div>
<div class="line"><a id="l00021" name="l00021"></a><span class="lineno"><a class="line" href="classmlx_1_1core_1_1distributed_1_1detail_1_1_group_impl.html#a65ae9485d2b1a2fd769744d50b0dd225"> 21</a></span> <span class="keyword">virtual</span> <span class="keywordtype">void</span> <a class="code hl_function" href="classmlx_1_1core_1_1distributed_1_1detail_1_1_group_impl.html#a65ae9485d2b1a2fd769744d50b0dd225">all_gather</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>&amp; input, <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>&amp; output, <a class="code hl_struct" href="structmlx_1_1core_1_1_stream.html">Stream</a> stream) = 0;</div>
<div class="line"><a id="l00022" name="l00022"></a><span class="lineno"><a class="line" href="classmlx_1_1core_1_1distributed_1_1detail_1_1_group_impl.html#a74befcdc600669cb87761106ae0bd9a5"> 22</a></span> <span class="keyword">virtual</span> <span class="keywordtype">void</span> <a class="code hl_function" href="classmlx_1_1core_1_1distributed_1_1detail_1_1_group_impl.html#a74befcdc600669cb87761106ae0bd9a5">send</a>(<span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>&amp; input, <span class="keywordtype">int</span> dst, <a class="code hl_struct" href="structmlx_1_1core_1_1_stream.html">Stream</a> stream) = 0;</div>
<div class="line"><a id="l00023" name="l00023"></a><span class="lineno"><a class="line" href="classmlx_1_1core_1_1distributed_1_1detail_1_1_group_impl.html#a7ce5b7a19d0fb8e189986c84845c5898"> 23</a></span> <span class="keyword">virtual</span> <span class="keywordtype">void</span> <a class="code hl_function" href="classmlx_1_1core_1_1distributed_1_1detail_1_1_group_impl.html#a7ce5b7a19d0fb8e189986c84845c5898">recv</a>(<a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>&amp; out, <span class="keywordtype">int</span> src, <a class="code hl_struct" href="structmlx_1_1core_1_1_stream.html">Stream</a> stream) = 0;</div>
<div class="line"><a id="l00024" name="l00024"></a><span class="lineno"> 24</span>};</div>
</div>
<div class="line"><a id="l00025" name="l00025"></a><span class="lineno"> 25</span> </div>
<div class="line"><a id="l00026" name="l00026"></a><span class="lineno"> 26</span><span class="comment">/* Return the communication stream. */</span></div>
<div class="line"><a id="l00027" name="l00027"></a><span class="lineno"><a class="line" href="namespacemlx_1_1core_1_1distributed_1_1detail.html#ac3612edf0e0e18c1e4ba0ce7c6e35cd6"> 27</a></span><a class="code hl_struct" href="structmlx_1_1core_1_1_stream.html">Stream</a> <a class="code hl_function" href="namespacemlx_1_1core_1_1distributed_1_1detail.html#ac3612edf0e0e18c1e4ba0ce7c6e35cd6">communication_stream</a>();</div>
<div class="line"><a id="l00026" name="l00026"></a><span class="lineno"> 26</span><span class="comment">/* Perform an all reduce sum operation */</span></div>
<div class="line"><a id="l00027" name="l00027"></a><span class="lineno"><a class="line" href="namespacemlx_1_1core_1_1distributed_1_1detail.html#a042be875217168ccfc267fba19a627cb"> 27</a></span><span class="keywordtype">void</span> <a class="code hl_function" href="namespacemlx_1_1core_1_1distributed_1_1detail.html#a042be875217168ccfc267fba19a627cb">all_sum</a>(<a class="code hl_struct" href="structmlx_1_1core_1_1distributed_1_1_group.html">Group</a> group, <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>&amp; input, <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>&amp; output, <a class="code hl_struct" href="structmlx_1_1core_1_1_stream.html">Stream</a> stream);</div>
<div class="line"><a id="l00028" name="l00028"></a><span class="lineno"> 28</span> </div>
<div class="line"><a id="l00029" name="l00029"></a><span class="lineno"> 29</span><span class="comment">/* Perform an all reduce sum operation */</span></div>
<div class="line"><a id="l00030" name="l00030"></a><span class="lineno"><a class="line" href="namespacemlx_1_1core_1_1distributed_1_1detail.html#aa1d225b25f7b6426c48c5e35860ee960"> 30</a></span><span class="keywordtype">void</span> <a class="code hl_function" href="namespacemlx_1_1core_1_1distributed_1_1detail.html#aa1d225b25f7b6426c48c5e35860ee960">all_sum</a>(<a class="code hl_struct" href="structmlx_1_1core_1_1distributed_1_1_group.html">Group</a> group, <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>&amp; input, <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>&amp; output);</div>
<div class="line"><a id="l00031" name="l00031"></a><span class="lineno"> 31</span> </div>
<div class="line"><a id="l00032" name="l00032"></a><span class="lineno"> 32</span><span class="comment">/* Perform an all gather operation */</span></div>
<div class="line"><a id="l00033" name="l00033"></a><span class="lineno"><a class="line" href="namespacemlx_1_1core_1_1distributed_1_1detail.html#aeb5a1726358213bc75756506f7b54d04"> 33</a></span><span class="keywordtype">void</span> <a class="code hl_function" href="namespacemlx_1_1core_1_1distributed_1_1detail.html#aeb5a1726358213bc75756506f7b54d04">all_gather</a>(<a class="code hl_struct" href="structmlx_1_1core_1_1distributed_1_1_group.html">Group</a> group, <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>&amp; input, <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>&amp; output);</div>
<div class="line"><a id="l00029" name="l00029"></a><span class="lineno"> 29</span><span class="comment">/* Perform an all gather operation */</span></div>
<div class="line"><a id="l00030" name="l00030"></a><span class="lineno"><a class="line" href="namespacemlx_1_1core_1_1distributed_1_1detail.html#ab3dc0367476257f13fe15d4db946edf5"> 30</a></span><span class="keywordtype">void</span> <a class="code hl_function" href="namespacemlx_1_1core_1_1distributed_1_1detail.html#ab3dc0367476257f13fe15d4db946edf5">all_gather</a>(<a class="code hl_struct" href="structmlx_1_1core_1_1distributed_1_1_group.html">Group</a> group, <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>&amp; input, <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>&amp; output, <a class="code hl_struct" href="structmlx_1_1core_1_1_stream.html">Stream</a> stream);</div>
<div class="line"><a id="l00031" name="l00031"></a><span class="lineno"> 31</span></div>
<div class="line"><a id="l00033" name="l00033"></a><span class="lineno"><a class="line" href="namespacemlx_1_1core_1_1distributed_1_1detail.html#a23c5cf992d4f2b2ce9dfa51593a4876d"> 33</a></span><span class="keywordtype">void</span> <a class="code hl_function" href="namespacemlx_1_1core_1_1distributed_1_1detail.html#a23c5cf992d4f2b2ce9dfa51593a4876d">send</a>(<a class="code hl_struct" href="structmlx_1_1core_1_1distributed_1_1_group.html">Group</a> group, <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>&amp; input, <span class="keywordtype">int</span> dst, <a class="code hl_struct" href="structmlx_1_1core_1_1_stream.html">Stream</a> stream);</div>
<div class="line"><a id="l00034" name="l00034"></a><span class="lineno"> 34</span></div>
<div class="line"><a id="l00036" name="l00036"></a><span class="lineno"><a class="line" href="namespacemlx_1_1core_1_1distributed_1_1detail.html#abf33511660ac71df5fc92f2aad6c6e08"> 36</a></span><span class="keywordtype">void</span> <a class="code hl_function" href="namespacemlx_1_1core_1_1distributed_1_1detail.html#abf33511660ac71df5fc92f2aad6c6e08">send</a>(<a class="code hl_struct" href="structmlx_1_1core_1_1distributed_1_1_group.html">Group</a> group, <span class="keyword">const</span> <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>&amp; input, <span class="keywordtype">int</span> dst);</div>
<div class="line"><a id="l00037" name="l00037"></a><span class="lineno"> 37</span></div>
<div class="line"><a id="l00039" name="l00039"></a><span class="lineno"><a class="line" href="namespacemlx_1_1core_1_1distributed_1_1detail.html#a003de04deb00ecbb19179b3f557df548"> 39</a></span><span class="keywordtype">void</span> <a class="code hl_function" href="namespacemlx_1_1core_1_1distributed_1_1detail.html#a003de04deb00ecbb19179b3f557df548">recv</a>(<a class="code hl_struct" href="structmlx_1_1core_1_1distributed_1_1_group.html">Group</a> group, <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>&amp; out, <span class="keywordtype">int</span> src);</div>
<div class="line"><a id="l00040" name="l00040"></a><span class="lineno"> 40</span> </div>
<div class="line"><a id="l00041" name="l00041"></a><span class="lineno"> 41</span>} <span class="comment">// namespace mlx::core::distributed::detail</span></div>
<div class="line"><a id="l00036" name="l00036"></a><span class="lineno"><a class="line" href="namespacemlx_1_1core_1_1distributed_1_1detail.html#a79bb934225482f2104e8ff270b0530c3"> 36</a></span><span class="keywordtype">void</span> <a class="code hl_function" href="namespacemlx_1_1core_1_1distributed_1_1detail.html#a79bb934225482f2104e8ff270b0530c3">recv</a>(<a class="code hl_struct" href="structmlx_1_1core_1_1distributed_1_1_group.html">Group</a> group, <a class="code hl_class" href="classmlx_1_1core_1_1array.html">array</a>&amp; out, <span class="keywordtype">int</span> src, <a class="code hl_struct" href="structmlx_1_1core_1_1_stream.html">Stream</a> stream);</div>
<div class="line"><a id="l00037" name="l00037"></a><span class="lineno"> 37</span> </div>
<div class="line"><a id="l00038" name="l00038"></a><span class="lineno"> 38</span>} <span class="comment">// namespace mlx::core::distributed::detail</span></div>
<div class="ttc" id="aclassmlx_1_1core_1_1array_html"><div class="ttname"><a href="classmlx_1_1core_1_1array.html">mlx::core::array</a></div><div class="ttdef"><b>Definition</b> array.h:24</div></div>
<div class="ttc" id="aclassmlx_1_1core_1_1distributed_1_1detail_1_1_group_impl_html"><div class="ttname"><a href="classmlx_1_1core_1_1distributed_1_1detail_1_1_group_impl.html">mlx::core::distributed::detail::GroupImpl</a></div><div class="ttdoc">Abstract base class of a distributed group implementation.</div><div class="ttdef"><b>Definition</b> distributed_impl.h:12</div></div>
<div class="ttc" id="aclassmlx_1_1core_1_1distributed_1_1detail_1_1_group_impl_html_a04bb1df23abe5b1f3fa0126375c6cea4"><div class="ttname"><a href="classmlx_1_1core_1_1distributed_1_1detail_1_1_group_impl.html#a04bb1df23abe5b1f3fa0126375c6cea4">mlx::core::distributed::detail::GroupImpl::all_gather</a></div><div class="ttdeci">virtual void all_gather(const array &amp;input, array &amp;output)=0</div></div>
<div class="ttc" id="aclassmlx_1_1core_1_1distributed_1_1detail_1_1_group_impl_html_a23f620ec55d75f236d5371e05a52fd64"><div class="ttname"><a href="classmlx_1_1core_1_1distributed_1_1detail_1_1_group_impl.html#a23f620ec55d75f236d5371e05a52fd64">mlx::core::distributed::detail::GroupImpl::all_sum</a></div><div class="ttdeci">virtual void all_sum(const array &amp;input, array &amp;output, Stream stream)=0</div></div>
<div class="ttc" id="aclassmlx_1_1core_1_1distributed_1_1detail_1_1_group_impl_html_a65ae9485d2b1a2fd769744d50b0dd225"><div class="ttname"><a href="classmlx_1_1core_1_1distributed_1_1detail_1_1_group_impl.html#a65ae9485d2b1a2fd769744d50b0dd225">mlx::core::distributed::detail::GroupImpl::all_gather</a></div><div class="ttdeci">virtual void all_gather(const array &amp;input, array &amp;output, Stream stream)=0</div></div>
<div class="ttc" id="aclassmlx_1_1core_1_1distributed_1_1detail_1_1_group_impl_html_a74befcdc600669cb87761106ae0bd9a5"><div class="ttname"><a href="classmlx_1_1core_1_1distributed_1_1detail_1_1_group_impl.html#a74befcdc600669cb87761106ae0bd9a5">mlx::core::distributed::detail::GroupImpl::send</a></div><div class="ttdeci">virtual void send(const array &amp;input, int dst, Stream stream)=0</div></div>
<div class="ttc" id="aclassmlx_1_1core_1_1distributed_1_1detail_1_1_group_impl_html_a7ce5b7a19d0fb8e189986c84845c5898"><div class="ttname"><a href="classmlx_1_1core_1_1distributed_1_1detail_1_1_group_impl.html#a7ce5b7a19d0fb8e189986c84845c5898">mlx::core::distributed::detail::GroupImpl::recv</a></div><div class="ttdeci">virtual void recv(array &amp;out, int src, Stream stream)=0</div></div>
<div class="ttc" id="aclassmlx_1_1core_1_1distributed_1_1detail_1_1_group_impl_html_a83f119b87210f53438c5ba675a78065e"><div class="ttname"><a href="classmlx_1_1core_1_1distributed_1_1detail_1_1_group_impl.html#a83f119b87210f53438c5ba675a78065e">mlx::core::distributed::detail::GroupImpl::~GroupImpl</a></div><div class="ttdeci">virtual ~GroupImpl()</div><div class="ttdef"><b>Definition</b> distributed_impl.h:14</div></div>
<div class="ttc" id="aclassmlx_1_1core_1_1distributed_1_1detail_1_1_group_impl_html_a87800a23c8160933a2d77a55a959194d"><div class="ttname"><a href="classmlx_1_1core_1_1distributed_1_1detail_1_1_group_impl.html#a87800a23c8160933a2d77a55a959194d">mlx::core::distributed::detail::GroupImpl::split</a></div><div class="ttdeci">virtual std::shared_ptr&lt; GroupImpl &gt; split(int color, int key=-1)=0</div></div>
<div class="ttc" id="aclassmlx_1_1core_1_1distributed_1_1detail_1_1_group_impl_html_ab1c8044b05f185c4bcc53002d4587599"><div class="ttname"><a href="classmlx_1_1core_1_1distributed_1_1detail_1_1_group_impl.html#ab1c8044b05f185c4bcc53002d4587599">mlx::core::distributed::detail::GroupImpl::size</a></div><div class="ttdeci">virtual int size()=0</div></div>
<div class="ttc" id="aclassmlx_1_1core_1_1distributed_1_1detail_1_1_group_impl_html_ac4af5fc16a82ba8c72df04d7694f8352"><div class="ttname"><a href="classmlx_1_1core_1_1distributed_1_1detail_1_1_group_impl.html#ac4af5fc16a82ba8c72df04d7694f8352">mlx::core::distributed::detail::GroupImpl::recv</a></div><div class="ttdeci">virtual void recv(array &amp;out, int src)=0</div></div>
<div class="ttc" id="aclassmlx_1_1core_1_1distributed_1_1detail_1_1_group_impl_html_ac8472eb2f96d1b14c7e4ccef56268ba0"><div class="ttname"><a href="classmlx_1_1core_1_1distributed_1_1detail_1_1_group_impl.html#ac8472eb2f96d1b14c7e4ccef56268ba0">mlx::core::distributed::detail::GroupImpl::send</a></div><div class="ttdeci">virtual void send(const array &amp;input, int dst)=0</div></div>
<div class="ttc" id="aclassmlx_1_1core_1_1distributed_1_1detail_1_1_group_impl_html_ae0838a40ce58442cdc73d57d7969a702"><div class="ttname"><a href="classmlx_1_1core_1_1distributed_1_1detail_1_1_group_impl.html#ae0838a40ce58442cdc73d57d7969a702">mlx::core::distributed::detail::GroupImpl::rank</a></div><div class="ttdeci">virtual int rank()=0</div></div>
<div class="ttc" id="aclassmlx_1_1core_1_1distributed_1_1detail_1_1_group_impl_html_ae163a6f444c6cc8820288b20f294e483"><div class="ttname"><a href="classmlx_1_1core_1_1distributed_1_1detail_1_1_group_impl.html#ae163a6f444c6cc8820288b20f294e483">mlx::core::distributed::detail::GroupImpl::all_sum</a></div><div class="ttdeci">virtual void all_sum(const array &amp;input, array &amp;output)=0</div></div>
<div class="ttc" id="adistributed_8h_html"><div class="ttname"><a href="distributed_8h.html">distributed.h</a></div></div>
<div class="ttc" id="anamespacemlx_1_1core_1_1distributed_1_1detail_html"><div class="ttname"><a href="namespacemlx_1_1core_1_1distributed_1_1detail.html">mlx::core::distributed::detail</a></div><div class="ttdef"><b>Definition</b> distributed.h:12</div></div>
<div class="ttc" id="anamespacemlx_1_1core_1_1distributed_1_1detail_html_a003de04deb00ecbb19179b3f557df548"><div class="ttname"><a href="namespacemlx_1_1core_1_1distributed_1_1detail.html#a003de04deb00ecbb19179b3f557df548">mlx::core::distributed::detail::recv</a></div><div class="ttdeci">void recv(Group group, array &amp;out, int src)</div><div class="ttdoc">Recv an array from the src rank.</div></div>
<div class="ttc" id="anamespacemlx_1_1core_1_1distributed_1_1detail_html_aa1d225b25f7b6426c48c5e35860ee960"><div class="ttname"><a href="namespacemlx_1_1core_1_1distributed_1_1detail.html#aa1d225b25f7b6426c48c5e35860ee960">mlx::core::distributed::detail::all_sum</a></div><div class="ttdeci">void all_sum(Group group, const array &amp;input, array &amp;output)</div></div>
<div class="ttc" id="anamespacemlx_1_1core_1_1distributed_1_1detail_html_abf33511660ac71df5fc92f2aad6c6e08"><div class="ttname"><a href="namespacemlx_1_1core_1_1distributed_1_1detail.html#abf33511660ac71df5fc92f2aad6c6e08">mlx::core::distributed::detail::send</a></div><div class="ttdeci">void send(Group group, const array &amp;input, int dst)</div><div class="ttdoc">Send an array to the dst rank.</div></div>
<div class="ttc" id="anamespacemlx_1_1core_1_1distributed_1_1detail_html_ac3612edf0e0e18c1e4ba0ce7c6e35cd6"><div class="ttname"><a href="namespacemlx_1_1core_1_1distributed_1_1detail.html#ac3612edf0e0e18c1e4ba0ce7c6e35cd6">mlx::core::distributed::detail::communication_stream</a></div><div class="ttdeci">Stream communication_stream()</div></div>
<div class="ttc" id="anamespacemlx_1_1core_1_1distributed_1_1detail_html_aeb5a1726358213bc75756506f7b54d04"><div class="ttname"><a href="namespacemlx_1_1core_1_1distributed_1_1detail.html#aeb5a1726358213bc75756506f7b54d04">mlx::core::distributed::detail::all_gather</a></div><div class="ttdeci">void all_gather(Group group, const array &amp;input, array &amp;output)</div></div>
<div class="ttc" id="anamespacemlx_1_1core_1_1distributed_1_1detail_html_a042be875217168ccfc267fba19a627cb"><div class="ttname"><a href="namespacemlx_1_1core_1_1distributed_1_1detail.html#a042be875217168ccfc267fba19a627cb">mlx::core::distributed::detail::all_sum</a></div><div class="ttdeci">void all_sum(Group group, const array &amp;input, array &amp;output, Stream stream)</div></div>
<div class="ttc" id="anamespacemlx_1_1core_1_1distributed_1_1detail_html_a23c5cf992d4f2b2ce9dfa51593a4876d"><div class="ttname"><a href="namespacemlx_1_1core_1_1distributed_1_1detail.html#a23c5cf992d4f2b2ce9dfa51593a4876d">mlx::core::distributed::detail::send</a></div><div class="ttdeci">void send(Group group, const array &amp;input, int dst, Stream stream)</div><div class="ttdoc">Send an array to the dst rank.</div></div>
<div class="ttc" id="anamespacemlx_1_1core_1_1distributed_1_1detail_html_a79bb934225482f2104e8ff270b0530c3"><div class="ttname"><a href="namespacemlx_1_1core_1_1distributed_1_1detail.html#a79bb934225482f2104e8ff270b0530c3">mlx::core::distributed::detail::recv</a></div><div class="ttdeci">void recv(Group group, array &amp;out, int src, Stream stream)</div><div class="ttdoc">Recv an array from the src rank.</div></div>
<div class="ttc" id="anamespacemlx_1_1core_1_1distributed_1_1detail_html_ab3dc0367476257f13fe15d4db946edf5"><div class="ttname"><a href="namespacemlx_1_1core_1_1distributed_1_1detail.html#ab3dc0367476257f13fe15d4db946edf5">mlx::core::distributed::detail::all_gather</a></div><div class="ttdeci">void all_gather(Group group, const array &amp;input, array &amp;output, Stream stream)</div></div>
<div class="ttc" id="astructmlx_1_1core_1_1_stream_html"><div class="ttname"><a href="structmlx_1_1core_1_1_stream.html">mlx::core::Stream</a></div><div class="ttdef"><b>Definition</b> stream.h:9</div></div>
<div class="ttc" id="astructmlx_1_1core_1_1distributed_1_1_group_html"><div class="ttname"><a href="structmlx_1_1core_1_1distributed_1_1_group.html">mlx::core::distributed::Group</a></div><div class="ttdoc">A distributed::Group represents a group of independent mlx processes that can communicate.</div><div class="ttdef"><b>Definition</b> distributed.h:24</div></div>
</div><!-- fragment --></div><!-- contents -->