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

30
docs/build/html/struct_g_e_m_v_kernel-members.html vendored
View File

@@ -102,25 +102,25 @@ $(function(){initNavTree('struct_g_e_m_v_kernel.html',''); initResizable(true);
</div>
<div class="header">
<div class="headertitle"><div class="title">GEMVKernel&lt; T, out_mask_t, op_mask_t, BM, BN, SM, SN, TM, TN &gt; Member List</div></div>
<div class="headertitle"><div class="title">GEMVKernel&lt; T, out_mask_t, op_mask_t, BM, BN, SM, SN, TM, TN, AccT &gt; Member List</div></div>
</div><!--header-->
<div class="contents">
<p>This is the complete list of members for <a class="el" href="struct_g_e_m_v_kernel.html">GEMVKernel&lt; T, out_mask_t, op_mask_t, BM, BN, SM, SN, TM, TN &gt;</a>, including all inherited members.</p>
<p>This is the complete list of members for <a class="el" href="struct_g_e_m_v_kernel.html">GEMVKernel&lt; T, out_mask_t, op_mask_t, BM, BN, SM, SN, TM, TN, AccT &gt;</a>, including all inherited members.</p>
<table class="directory">
<tr class="even"><td class="entry"><a class="el" href="struct_g_e_m_v_kernel.html#a7281520100658811076400060663903c">blockM</a></td><td class="entry"><a class="el" href="struct_g_e_m_v_kernel.html">GEMVKernel&lt; T, out_mask_t, op_mask_t, BM, BN, SM, SN, TM, TN &gt;</a></td><td class="entry"><span class="mlabel">static</span></td></tr>
<tr class="odd"><td class="entry"><a class="el" href="struct_g_e_m_v_kernel.html#a2fef17f9c9aa0bdf530ad3554fb0988b">blockN</a></td><td class="entry"><a class="el" href="struct_g_e_m_v_kernel.html">GEMVKernel&lt; T, out_mask_t, op_mask_t, BM, BN, SM, SN, TM, TN &gt;</a></td><td class="entry"><span class="mlabel">static</span></td></tr>
<tr class="even"><td class="entry"><a class="el" href="struct_g_e_m_v_kernel.html#ad47223ee49b3cb7bf3746a2cec45f883">has_mul_operand_mask</a></td><td class="entry"><a class="el" href="struct_g_e_m_v_kernel.html">GEMVKernel&lt; T, out_mask_t, op_mask_t, BM, BN, SM, SN, TM, TN &gt;</a></td><td class="entry"><span class="mlabel">static</span></td></tr>
<tr class="odd"><td class="entry"><a class="el" href="struct_g_e_m_v_kernel.html#a0edbf2dd6a6563e7afa6dab6b670615c">has_mul_output_mask</a></td><td class="entry"><a class="el" href="struct_g_e_m_v_kernel.html">GEMVKernel&lt; T, out_mask_t, op_mask_t, BM, BN, SM, SN, TM, TN &gt;</a></td><td class="entry"><span class="mlabel">static</span></td></tr>
<tr class="even"><td class="entry"><a class="el" href="struct_g_e_m_v_kernel.html#ab00784dff1512a7b0919fcb4cfa5d50e">has_operand_mask</a></td><td class="entry"><a class="el" href="struct_g_e_m_v_kernel.html">GEMVKernel&lt; T, out_mask_t, op_mask_t, BM, BN, SM, SN, TM, TN &gt;</a></td><td class="entry"><span class="mlabel">static</span></td></tr>
<tr class="odd"><td class="entry"><a class="el" href="struct_g_e_m_v_kernel.html#ab8b64c94f4c8f6f09c0777415589b487">has_output_mask</a></td><td class="entry"><a class="el" href="struct_g_e_m_v_kernel.html">GEMVKernel&lt; T, out_mask_t, op_mask_t, BM, BN, SM, SN, TM, TN &gt;</a></td><td class="entry"><span class="mlabel">static</span></td></tr>
<tr class="even"><td class="entry"><a class="el" href="struct_g_e_m_v_kernel.html#a04bb72da9a93d6d1eba468fa311bbba7">load_safe</a>(const device T *src, thread T dst[TN], const int src_offset=0, const int src_size=TN)</td><td class="entry"><a class="el" href="struct_g_e_m_v_kernel.html">GEMVKernel&lt; T, out_mask_t, op_mask_t, BM, BN, SM, SN, TM, TN &gt;</a></td><td class="entry"><span class="mlabel">inline</span><span class="mlabel">static</span></td></tr>
<tr class="odd"><td class="entry"><a class="el" href="struct_g_e_m_v_kernel.html#a6013e9c5b2f72fa1311dd038172df0ce">load_unsafe</a>(const device T *src, thread T dst[TN], const int src_offset=0)</td><td class="entry"><a class="el" href="struct_g_e_m_v_kernel.html">GEMVKernel&lt; T, out_mask_t, op_mask_t, BM, BN, SM, SN, TM, TN &gt;</a></td><td class="entry"><span class="mlabel">inline</span><span class="mlabel">static</span></td></tr>
<tr class="even"><td class="entry"><a class="el" href="struct_g_e_m_v_kernel.html#ae8113fddf6fb637acfd12efd978b704c">needs_tgp_reduction</a></td><td class="entry"><a class="el" href="struct_g_e_m_v_kernel.html">GEMVKernel&lt; T, out_mask_t, op_mask_t, BM, BN, SM, SN, TM, TN &gt;</a></td><td class="entry"><span class="mlabel">static</span></td></tr>
<tr class="odd"><td class="entry"><a class="el" href="struct_g_e_m_v_kernel.html#ac4a7b5011a0ea938ab1949bb1767fc1a">run</a>(const device T *mat, const device T *in_vec, device T *out_vec, const constant int &amp;in_vec_size, const constant int &amp;out_vec_size, const constant int &amp;matrix_ld, const device out_mask_t *out_mask, const device op_mask_t *mat_mask, const device op_mask_t *vec_mask, const constant int *mask_strides, threadgroup T *tgp_memory, uint3 tid, uint3 lid, uint simd_gid, uint simd_lid)</td><td class="entry"><a class="el" href="struct_g_e_m_v_kernel.html">GEMVKernel&lt; T, out_mask_t, op_mask_t, BM, BN, SM, SN, TM, TN &gt;</a></td><td class="entry"><span class="mlabel">inline</span><span class="mlabel">static</span></td></tr>
<tr class="even"><td class="entry"><a class="el" href="struct_g_e_m_v_kernel.html#a9ef4d0e62094d7033069f5dda5efb236">tgp_mem_size</a></td><td class="entry"><a class="el" href="struct_g_e_m_v_kernel.html">GEMVKernel&lt; T, out_mask_t, op_mask_t, BM, BN, SM, SN, TM, TN &gt;</a></td><td class="entry"><span class="mlabel">static</span></td></tr>
<tr class="odd"><td class="entry"><a class="el" href="struct_g_e_m_v_kernel.html#a1dd943fcbf5e7be435fc36bed589a641">threadsM</a></td><td class="entry"><a class="el" href="struct_g_e_m_v_kernel.html">GEMVKernel&lt; T, out_mask_t, op_mask_t, BM, BN, SM, SN, TM, TN &gt;</a></td><td class="entry"><span class="mlabel">static</span></td></tr>
<tr class="even"><td class="entry"><a class="el" href="struct_g_e_m_v_kernel.html#a47bfab7d21dd18760d3e0937ad36b19d">threadsN</a></td><td class="entry"><a class="el" href="struct_g_e_m_v_kernel.html">GEMVKernel&lt; T, out_mask_t, op_mask_t, BM, BN, SM, SN, TM, TN &gt;</a></td><td class="entry"><span class="mlabel">static</span></td></tr>
<tr class="even"><td class="entry"><a class="el" href="struct_g_e_m_v_kernel.html#a188ef7ed5c1d74d9b540b6d4ebf12f2e">blockM</a></td><td class="entry"><a class="el" href="struct_g_e_m_v_kernel.html">GEMVKernel&lt; T, out_mask_t, op_mask_t, BM, BN, SM, SN, TM, TN, AccT &gt;</a></td><td class="entry"><span class="mlabel">static</span></td></tr>
<tr class="odd"><td class="entry"><a class="el" href="struct_g_e_m_v_kernel.html#af22b6e5ed1a9d350866aaafa35d63a8a">blockN</a></td><td class="entry"><a class="el" href="struct_g_e_m_v_kernel.html">GEMVKernel&lt; T, out_mask_t, op_mask_t, BM, BN, SM, SN, TM, TN, AccT &gt;</a></td><td class="entry"><span class="mlabel">static</span></td></tr>
<tr class="even"><td class="entry"><a class="el" href="struct_g_e_m_v_kernel.html#a09f0e646822d45dc2543e31509552258">has_mul_operand_mask</a></td><td class="entry"><a class="el" href="struct_g_e_m_v_kernel.html">GEMVKernel&lt; T, out_mask_t, op_mask_t, BM, BN, SM, SN, TM, TN, AccT &gt;</a></td><td class="entry"><span class="mlabel">static</span></td></tr>
<tr class="odd"><td class="entry"><a class="el" href="struct_g_e_m_v_kernel.html#ad54fb5c6f0f9a820365b638542693291">has_mul_output_mask</a></td><td class="entry"><a class="el" href="struct_g_e_m_v_kernel.html">GEMVKernel&lt; T, out_mask_t, op_mask_t, BM, BN, SM, SN, TM, TN, AccT &gt;</a></td><td class="entry"><span class="mlabel">static</span></td></tr>
<tr class="even"><td class="entry"><a class="el" href="struct_g_e_m_v_kernel.html#a68bad880d1e689228ae4d3c6958cc6c1">has_operand_mask</a></td><td class="entry"><a class="el" href="struct_g_e_m_v_kernel.html">GEMVKernel&lt; T, out_mask_t, op_mask_t, BM, BN, SM, SN, TM, TN, AccT &gt;</a></td><td class="entry"><span class="mlabel">static</span></td></tr>
<tr class="odd"><td class="entry"><a class="el" href="struct_g_e_m_v_kernel.html#a3b0b9ccf11bd5d7de50b9626fa9a22cc">has_output_mask</a></td><td class="entry"><a class="el" href="struct_g_e_m_v_kernel.html">GEMVKernel&lt; T, out_mask_t, op_mask_t, BM, BN, SM, SN, TM, TN, AccT &gt;</a></td><td class="entry"><span class="mlabel">static</span></td></tr>
<tr class="even"><td class="entry"><a class="el" href="struct_g_e_m_v_kernel.html#a047eca250e7cbc928bce0e13de98e558">load_safe</a>(const device T *src, thread U dst[TN], const int src_offset=0, const int src_size=TN)</td><td class="entry"><a class="el" href="struct_g_e_m_v_kernel.html">GEMVKernel&lt; T, out_mask_t, op_mask_t, BM, BN, SM, SN, TM, TN, AccT &gt;</a></td><td class="entry"><span class="mlabel">inline</span><span class="mlabel">static</span></td></tr>
<tr class="odd"><td class="entry"><a class="el" href="struct_g_e_m_v_kernel.html#af96a768a213a95e7a4d8f3ec1948034f">load_unsafe</a>(const device T *src, thread U dst[TN], const int src_offset=0)</td><td class="entry"><a class="el" href="struct_g_e_m_v_kernel.html">GEMVKernel&lt; T, out_mask_t, op_mask_t, BM, BN, SM, SN, TM, TN, AccT &gt;</a></td><td class="entry"><span class="mlabel">inline</span><span class="mlabel">static</span></td></tr>
<tr class="even"><td class="entry"><a class="el" href="struct_g_e_m_v_kernel.html#ae1c8e57e6718f22732570cc92d9bcf99">needs_tgp_reduction</a></td><td class="entry"><a class="el" href="struct_g_e_m_v_kernel.html">GEMVKernel&lt; T, out_mask_t, op_mask_t, BM, BN, SM, SN, TM, TN, AccT &gt;</a></td><td class="entry"><span class="mlabel">static</span></td></tr>
<tr class="odd"><td class="entry"><a class="el" href="struct_g_e_m_v_kernel.html#ac1a9e1d9853489dd928916912cc627a7">run</a>(const device T *mat, const device T *in_vec, device T *out_vec, const constant int &amp;in_vec_size, const constant int &amp;out_vec_size, const constant int &amp;matrix_ld, const device out_mask_t *out_mask, const device op_mask_t *mat_mask, const device op_mask_t *vec_mask, const constant int *mask_strides, threadgroup AccT *tgp_memory, uint3 tid, uint3 lid, uint simd_gid, uint simd_lid)</td><td class="entry"><a class="el" href="struct_g_e_m_v_kernel.html">GEMVKernel&lt; T, out_mask_t, op_mask_t, BM, BN, SM, SN, TM, TN, AccT &gt;</a></td><td class="entry"><span class="mlabel">inline</span><span class="mlabel">static</span></td></tr>
<tr class="even"><td class="entry"><a class="el" href="struct_g_e_m_v_kernel.html#a53514fa199efc5b7328052dac45aabab">tgp_mem_size</a></td><td class="entry"><a class="el" href="struct_g_e_m_v_kernel.html">GEMVKernel&lt; T, out_mask_t, op_mask_t, BM, BN, SM, SN, TM, TN, AccT &gt;</a></td><td class="entry"><span class="mlabel">static</span></td></tr>
<tr class="odd"><td class="entry"><a class="el" href="struct_g_e_m_v_kernel.html#afbe7ab8ebfa912ffbf3ba2cf4db24940">threadsM</a></td><td class="entry"><a class="el" href="struct_g_e_m_v_kernel.html">GEMVKernel&lt; T, out_mask_t, op_mask_t, BM, BN, SM, SN, TM, TN, AccT &gt;</a></td><td class="entry"><span class="mlabel">static</span></td></tr>
<tr class="even"><td class="entry"><a class="el" href="struct_g_e_m_v_kernel.html#a23142b7400f1f678e5d6d69d87f51032">threadsN</a></td><td class="entry"><a class="el" href="struct_g_e_m_v_kernel.html">GEMVKernel&lt; T, out_mask_t, op_mask_t, BM, BN, SM, SN, TM, TN, AccT &gt;</a></td><td class="entry"><span class="mlabel">static</span></td></tr>
</table></div><!-- contents -->
</div><!-- doc-content -->
<!-- start footer part -->