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

rebase

Browse Source
This commit is contained in:
CircleCI Docs
2024-11-22 20:24:16 +00:00
parent 6433a22733
commit 8ba8a39e43
905 changed files with 30035 additions and 16934 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
docs/build/html/sort_8h_source.html vendored
View File

@@ -464,8 +464,8 @@ $(function(){ initResizable(false); });
<div class="line"><a id="l00352" name="l00352"></a><span class="lineno"> 352</span> <span class="keyword">using </span>val_t = <span class="keyword">typename</span> sort_kernel::val_t;</div>
<div class="line"><a id="l00353" name="l00353"></a><span class="lineno"> 353</span> <span class="keyword">using </span>idx_t = <span class="keyword">typename</span> sort_kernel::idx_t;</div>
<div class="line"><a id="l00354" name="l00354"></a><span class="lineno"> 354</span> </div>
<div class="line"><a id="l00355" name="l00355"></a><span class="lineno"> 355</span> <span class="keyword">auto</span> in_block_idx = <a class="code hl_function" href="backend_2metal_2kernels_2utils_8h.html#a8fd0c8fc6058e650fc99bca8b6acd7d1">elem_to_loc</a>(tid.y, nc_shape, in_nc_strides, nc_dim);</div>
<div class="line"><a id="l00356" name="l00356"></a><span class="lineno"> 356</span> <span class="keyword">auto</span> out_block_idx = <a class="code hl_function" href="backend_2metal_2kernels_2utils_8h.html#a8fd0c8fc6058e650fc99bca8b6acd7d1">elem_to_loc</a>(tid.y, nc_shape, out_nc_strides, nc_dim);</div>
<div class="line"><a id="l00355" name="l00355"></a><span class="lineno"> 355</span> <span class="keyword">auto</span> in_block_idx = <a class="code hl_function" href="backend_2metal_2kernels_2utils_8h.html#a22eaa505dbc7dd2a63a895f2e16712f5">elem_to_loc</a>(tid.y, nc_shape, in_nc_strides, nc_dim);</div>
<div class="line"><a id="l00356" name="l00356"></a><span class="lineno"> 356</span> <span class="keyword">auto</span> out_block_idx = <a class="code hl_function" href="backend_2metal_2kernels_2utils_8h.html#a22eaa505dbc7dd2a63a895f2e16712f5">elem_to_loc</a>(tid.y, nc_shape, out_nc_strides, nc_dim);</div>
<div class="line"><a id="l00357" name="l00357"></a><span class="lineno"> 357</span> inp += in_block_idx;</div>
<div class="line"><a id="l00358" name="l00358"></a><span class="lineno"> 358</span> out += out_block_idx;</div>
<div class="line"><a id="l00359" name="l00359"></a><span class="lineno"> 359</span> </div>
@@ -616,7 +616,7 @@ $(function(){ initResizable(false); });
<div class="line"><a id="l00496" name="l00496"></a><span class="lineno"> 496</span> BLOCK_THREADS,</div>
<div class="line"><a id="l00497" name="l00497"></a><span class="lineno"> 497</span> N_PER_THREAD&gt;;</div>
<div class="line"><a id="l00498" name="l00498"></a><span class="lineno"> 498</span> </div>
<div class="line"><a id="l00499" name="l00499"></a><span class="lineno"> 499</span> <span class="keyword">auto</span> block_idx = <a class="code hl_function" href="backend_2metal_2kernels_2utils_8h.html#a8fd0c8fc6058e650fc99bca8b6acd7d1">elem_to_loc</a>(tid.y, nc_shape, nc_strides, nc_dim);</div>
<div class="line"><a id="l00499" name="l00499"></a><span class="lineno"> 499</span> <span class="keyword">auto</span> block_idx = <a class="code hl_function" href="backend_2metal_2kernels_2utils_8h.html#a22eaa505dbc7dd2a63a895f2e16712f5">elem_to_loc</a>(tid.y, nc_shape, nc_strides, nc_dim);</div>
<div class="line"><a id="l00500" name="l00500"></a><span class="lineno"> 500</span> inp += block_idx;</div>
<div class="line"><a id="l00501" name="l00501"></a><span class="lineno"> 501</span> out_vals += tid.y * size_sorted_axis;</div>
<div class="line"><a id="l00502" name="l00502"></a><span class="lineno"> 502</span> out_idxs += tid.y * size_sorted_axis;</div>
@@ -818,11 +818,11 @@ $(function(){ initResizable(false); });
<div class="line"><a id="l00694" name="l00694"></a><span class="lineno"> 694</span> }</div>
<div class="line"><a id="l00695" name="l00695"></a><span class="lineno"> 695</span>}</div>
</div>
<div class="ttc" id="abackend_2metal_2kernels_2utils_8h_html_a8fd0c8fc6058e650fc99bca8b6acd7d1"><div class="ttname"><a href="backend_2metal_2kernels_2utils_8h.html#a8fd0c8fc6058e650fc99bca8b6acd7d1">elem_to_loc</a></div><div class="ttdeci">METAL_FUNC stride_t elem_to_loc(uint elem, constant const int *shape, constant const stride_t *strides, int ndim)</div><div class="ttdef"><b>Definition</b> utils.h:87</div></div>
<div class="ttc" id="abackend_2metal_2kernels_2utils_8h_html_a22eaa505dbc7dd2a63a895f2e16712f5"><div class="ttname"><a href="backend_2metal_2kernels_2utils_8h.html#a22eaa505dbc7dd2a63a895f2e16712f5">elem_to_loc</a></div><div class="ttdeci">METAL_FUNC IdxT elem_to_loc(uint elem, constant const int *shape, constant const StrideT *strides, int ndim)</div><div class="ttdef"><b>Definition</b> utils.h:93</div></div>
<div class="ttc" id="acommon_2binary_8h_html_a70228731d29946574b238d21fb4b360c"><div class="ttname"><a href="common_2binary_8h.html#a70228731d29946574b238d21fb4b360c">op</a></div><div class="ttdeci">Op op</div><div class="ttdef"><b>Definition</b> binary.h:129</div></div>
<div class="ttc" id="anamespacemetal_html"><div class="ttname"><a href="namespacemetal.html">metal</a></div><div class="ttdef"><b>Definition</b> bf16.h:265</div></div>
<div class="ttc" id="anamespacemetal_html_a6653b28c9473087141eddce39878d4d3"><div class="ttname"><a href="namespacemetal.html#a6653b28c9473087141eddce39878d4d3">metal::min</a></div><div class="ttdeci">METAL_FUNC bfloat16_t min(bfloat16_t x, bfloat16_t y)</div><div class="ttdef"><b>Definition</b> bf16_math.h:234</div></div>
<div class="ttc" id="anamespacemetal_html_a853c80479ab2264d9c4587c7bcac767b"><div class="ttname"><a href="namespacemetal.html#a853c80479ab2264d9c4587c7bcac767b">metal::max</a></div><div class="ttdeci">METAL_FUNC bfloat16_t max(bfloat16_t x, bfloat16_t y)</div><div class="ttdef"><b>Definition</b> bf16_math.h:234</div></div>
<div class="ttc" id="anamespacemetal_html"><div class="ttname"><a href="namespacemetal.html">metal</a></div><div class="ttdef"><b>Definition</b> bf16_math.h:226</div></div>
<div class="ttc" id="anamespacemetal_html_a6653b28c9473087141eddce39878d4d3"><div class="ttname"><a href="namespacemetal.html#a6653b28c9473087141eddce39878d4d3">metal::min</a></div><div class="ttdeci">METAL_FUNC bfloat16_t min(bfloat16_t x, bfloat16_t y)</div><div class="ttdef"><b>Definition</b> bf16_math.h:232</div></div>
<div class="ttc" id="anamespacemetal_html_a853c80479ab2264d9c4587c7bcac767b"><div class="ttname"><a href="namespacemetal.html#a853c80479ab2264d9c4587c7bcac767b">metal::max</a></div><div class="ttdeci">METAL_FUNC bfloat16_t max(bfloat16_t x, bfloat16_t y)</div><div class="ttdef"><b>Definition</b> bf16_math.h:232</div></div>
<div class="ttc" id="asort_8h_html_a0386011c52d03e60885a31e6fbd903dd"><div class="ttname"><a href="sort_8h.html#a0386011c52d03e60885a31e6fbd903dd">MLX_MTL_CONST</a></div><div class="ttdeci">#define MLX_MTL_CONST</div><div class="ttdef"><b>Definition</b> sort.h:3</div></div>
<div class="ttc" id="asort_8h_html_a32cbe4163b8b0f5cb2c97b256119a4b2"><div class="ttname"><a href="sort_8h.html#a32cbe4163b8b0f5cb2c97b256119a4b2">mb_block_partition</a></div><div class="ttdeci">void mb_block_partition(device idx_t *block_partitions, const device val_t *dev_vals, const device idx_t *dev_idxs, const constant int &amp;size_sorted_axis, const constant int &amp;merge_tiles, const constant int &amp;n_blocks, uint3 tid, uint3 lid, uint3 tgp_dims)</div><div class="ttdef"><b>Definition</b> sort.h:525</div></div>
<div class="ttc" id="asort_8h_html_a4ee3de195a6f9c33aa91ac52461808ad"><div class="ttname"><a href="sort_8h.html#a4ee3de195a6f9c33aa91ac52461808ad">block_sort_nc</a></div><div class="ttdeci">void block_sort_nc(const device T *inp, device U *out, const constant int &amp;size_sorted_axis, const constant int &amp;in_stride_sorted_axis, const constant int &amp;out_stride_sorted_axis, const constant int &amp;nc_dim, const constant int *nc_shape, const constant size_t *in_nc_strides, const constant size_t *out_nc_strides, uint3 tid, uint3 lid)</div><div class="ttdef"><b>Definition</b> sort.h:338</div></div>
@@ -848,7 +848,7 @@ $(function(){ initResizable(false); });
<div class="ttc" id="astruct_less_than_html"><div class="ttname"><a href="struct_less_than.html">LessThan</a></div><div class="ttdef"><b>Definition</b> sort.h:23</div></div>
<div class="ttc" id="astruct_less_than_html_a2798eb377b411c93a4ed30cf35caade2"><div class="ttname"><a href="struct_less_than.html#a2798eb377b411c93a4ed30cf35caade2">LessThan::operator()</a></div><div class="ttdeci">METAL_FUNC bool operator()(T a, T b)</div><div class="ttdef"><b>Definition</b> sort.h:26</div></div>
<div class="ttc" id="astruct_less_than_html_abf97a6b0163048e4ba96460939dbd3a3"><div class="ttname"><a href="struct_less_than.html#abf97a6b0163048e4ba96460939dbd3a3">LessThan::init</a></div><div class="ttdeci">static constexpr constant T init</div><div class="ttdef"><b>Definition</b> sort.h:24</div></div>
<div class="ttc" id="astruct_limits_html"><div class="ttname"><a href="struct_limits.html">Limits</a></div><div class="ttdef"><b>Definition</b> utils.h:17</div></div>
<div class="ttc" id="astruct_limits_html"><div class="ttname"><a href="struct_limits.html">Limits</a></div><div class="ttdef"><b>Definition</b> utils.h:23</div></div>
<div class="ttc" id="astruct_thread_sort_html"><div class="ttname"><a href="struct_thread_sort.html">ThreadSort</a></div><div class="ttdef"><b>Definition</b> sort.h:37</div></div>
<div class="ttc" id="astruct_thread_sort_html_ad9ab3e6b47f7e9b91c0f3b773596986d"><div class="ttname"><a href="struct_thread_sort.html#ad9ab3e6b47f7e9b91c0f3b773596986d">ThreadSort::sort</a></div><div class="ttdeci">static METAL_FUNC void sort(thread val_t(&amp;vals)[N_PER_THREAD], thread idx_t(&amp;idxs)[N_PER_THREAD])</div><div class="ttdef"><b>Definition</b> sort.h:38</div></div>
</div><!-- fragment --></div><!-- contents -->