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

rebase

Browse Source
This commit is contained in:
CircleCI Docs
2025-03-05 21:30:09 +00:00
parent 75441e48ef
commit a943912d4c
733 changed files with 41418 additions and 30412 deletions
Download Patch File Download Diff File Expand all files Collapse all files

38
docs/build/html/namespacemlx_1_1steel.html vendored
View File

@@ -205,6 +205,9 @@ Typedefs</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:ab9fdcb06fb1f639f9120ab14cfedd150" id="r_ab9fdcb06fb1f639f9120ab14cfedd150"><td class="memTemplParams" colspan="2">template&lt;typename Dtype, typename Atype, typename Btype, typename Ctype, int M, int N, int K, class MMAFragD, class MMAFragA, class MMAFragB, class MMAFragC&gt; </td></tr>
<tr class="memitem:ab9fdcb06fb1f639f9120ab14cfedd150"><td class="memTemplItemLeft" align="right" valign="top">METAL_FUNC void&#160;</td><td class="memTemplItemRight" valign="bottom"><a class="el" href="#ab9fdcb06fb1f639f9120ab14cfedd150">tile_matmad</a> (thread <a class="el" href="structmlx_1_1steel_1_1_m_m_a_tile.html">MMATile</a>&lt; Dtype, M, N, MMAFragD &gt; &amp;D, thread <a class="el" href="structmlx_1_1steel_1_1_m_m_a_tile.html">MMATile</a>&lt; Atype, M, K, MMAFragA &gt; &amp;A, thread <a class="el" href="structmlx_1_1steel_1_1_m_m_a_tile.html">MMATile</a>&lt; Btype, K, N, MMAFragB &gt; &amp;B, thread <a class="el" href="structmlx_1_1steel_1_1_m_m_a_tile.html">MMATile</a>&lt; Ctype, M, N, MMAFragC &gt; &amp;C)</td></tr>
<tr class="separator:ab9fdcb06fb1f639f9120ab14cfedd150"><td class="memSeparator" colspan="2">&#160;</td></tr>
<tr class="memitem:ad583e6038efc119542410f43b603d4ad" id="r_ad583e6038efc119542410f43b603d4ad"><td class="memTemplParams" colspan="2">template&lt;typename T, typename U, int M, int N, int K&gt; </td></tr>
<tr class="memitem:ad583e6038efc119542410f43b603d4ad"><td class="memTemplItemLeft" align="right" valign="top">METAL_FUNC void&#160;</td><td class="memTemplItemRight" valign="bottom"><a class="el" href="#ad583e6038efc119542410f43b603d4ad">tile_matmad</a> (thread <a class="el" href="structmlx_1_1steel_1_1_m_m_a_tile.html">MMATile</a>&lt; T, M, N &gt; &amp;D, thread <a class="el" href="structmlx_1_1steel_1_1_m_m_a_tile.html">MMATile</a>&lt; U, M, K &gt; &amp;A, thread <a class="el" href="structmlx_1_1steel_1_1_m_m_a_tile.html">MMATile</a>&lt; U, K, N &gt; &amp;B, thread <a class="el" href="structmlx_1_1steel_1_1_m_m_a_tile.html">MMATile</a>&lt; T, M, N &gt; &amp;C)</td></tr>
<tr class="separator:ad583e6038efc119542410f43b603d4ad"><td class="memSeparator" colspan="2">&#160;</td></tr>
@@ -747,10 +750,43 @@ template&lt;typename T, typename... Us&gt; </div>
</table>
</div><div class="memdoc">
</div>
</div>
<a id="ab9fdcb06fb1f639f9120ab14cfedd150" name="ab9fdcb06fb1f639f9120ab14cfedd150"></a>
<h2 class="memtitle"><span class="permalink"><a href="#ab9fdcb06fb1f639f9120ab14cfedd150">&#9670;&#160;</a></span>tile_matmad() <span class="overload">[1/2]</span></h2>
<div class="memitem">
<div class="memproto">
<div class="memtemplate">
template&lt;typename Dtype, typename Atype, typename Btype, typename Ctype, int M, int N, int K, class MMAFragD, class MMAFragA, class MMAFragB, class MMAFragC&gt; </div>
<table class="memname">
<tr>
<td class="memname">METAL_FUNC void mlx::steel::tile_matmad </td>
<td>(</td>
<td class="paramtype">thread <a class="el" href="structmlx_1_1steel_1_1_m_m_a_tile.html">MMATile</a>&lt; Dtype, M, N, MMAFragD &gt; &amp;</td> <td class="paramname"><span class="paramname"><em>D</em></span>, </td>
</tr>
<tr>
<td class="paramkey"></td>
<td></td>
<td class="paramtype">thread <a class="el" href="structmlx_1_1steel_1_1_m_m_a_tile.html">MMATile</a>&lt; Atype, M, K, MMAFragA &gt; &amp;</td> <td class="paramname"><span class="paramname"><em>A</em></span>, </td>
</tr>
<tr>
<td class="paramkey"></td>
<td></td>
<td class="paramtype">thread <a class="el" href="structmlx_1_1steel_1_1_m_m_a_tile.html">MMATile</a>&lt; Btype, K, N, MMAFragB &gt; &amp;</td> <td class="paramname"><span class="paramname"><em>B</em></span>, </td>
</tr>
<tr>
<td class="paramkey"></td>
<td></td>
<td class="paramtype">thread <a class="el" href="structmlx_1_1steel_1_1_m_m_a_tile.html">MMATile</a>&lt; Ctype, M, N, MMAFragC &gt; &amp;</td> <td class="paramname"><span class="paramname"><em>C</em></span>&#160;)</td>
</tr>
</table>
</div><div class="memdoc">
</div>
</div>
<a id="ad583e6038efc119542410f43b603d4ad" name="ad583e6038efc119542410f43b603d4ad"></a>
<h2 class="memtitle"><span class="permalink"><a href="#ad583e6038efc119542410f43b603d4ad">&#9670;&#160;</a></span>tile_matmad()</h2>
<h2 class="memtitle"><span class="permalink"><a href="#ad583e6038efc119542410f43b603d4ad">&#9670;&#160;</a></span>tile_matmad() <span class="overload">[2/2]</span></h2>
<div class="memitem">
<div class="memproto">