zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-12-16 01:49:05 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

rebase

Browse Source
This commit is contained in:
CircleCI Docs
2024-12-06 21:22:39 +00:00
parent 8ba8a39e43
commit 04b749a588
896 changed files with 12293 additions and 11212 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
docs/build/html/metal_2kernels_2binary__two_8h_source.html vendored
View File

@@ -197,9 +197,9 @@ $(function(){ initResizable(false); });
<div class="line"><a id="l00091" name="l00091"></a><span class="lineno"> 91</span>}</div>
</div>
<div class="line"><a id="l00092" name="l00092"></a><span class="lineno"> 92</span> </div>
<div class="line"><a id="l00093" name="l00093"></a><span class="lineno"> 93</span><span class="keyword">template</span> &lt;<span class="keyword">typename</span> T, <span class="keyword">typename</span> U, <span class="keyword">typename</span> Op&gt;</div>
<div class="line"><a id="l00093" name="l00093"></a><span class="lineno"> 93</span><span class="keyword">template</span> &lt;<span class="keyword">typename</span> T, <span class="keyword">typename</span> U, <span class="keyword">typename</span> Op, <span class="keyword">typename</span> IdxT = <span class="keywordtype">size_t</span>&gt;</div>
<div class="foldopen" id="foldopen00094" data-start="{" data-end="}">
<div class="line"><a id="l00094" name="l00094"></a><span class="lineno"><a class="line" href="metal_2kernels_2binary__two_8h.html#ad1fad37c168192b212a4294f4cf78133"> 94</a></span>[[kernel]] <span class="keywordtype">void</span> <a class="code hl_function" href="metal_2kernels_2binary__two_8h.html#ad1fad37c168192b212a4294f4cf78133">binary_g_nd1</a>(</div>
<div class="line"><a id="l00094" name="l00094"></a><span class="lineno"><a class="line" href="metal_2kernels_2binary__two_8h.html#a44e8241fa805906fa9644204e4d1237f"> 94</a></span>[[kernel]] <span class="keywordtype">void</span> <a class="code hl_function" href="metal_2kernels_2binary__two_8h.html#a44e8241fa805906fa9644204e4d1237f">binary_g_nd1</a>(</div>
<div class="line"><a id="l00095" name="l00095"></a><span class="lineno"> 95</span> device <span class="keyword">const</span> T* a,</div>
<div class="line"><a id="l00096" name="l00096"></a><span class="lineno"> 96</span> device <span class="keyword">const</span> T* b,</div>
<div class="line"><a id="l00097" name="l00097"></a><span class="lineno"> 97</span> device U* c,</div>
@@ -207,8 +207,8 @@ $(function(){ initResizable(false); });
<div class="line"><a id="l00099" name="l00099"></a><span class="lineno"> 99</span> constant <span class="keyword">const</span> <span class="keywordtype">size_t</span>&amp; a_stride,</div>
<div class="line"><a id="l00100" name="l00100"></a><span class="lineno"> 100</span> constant <span class="keyword">const</span> <span class="keywordtype">size_t</span>&amp; b_stride,</div>
<div class="line"><a id="l00101" name="l00101"></a><span class="lineno"> 101</span> uint index [[thread_position_in_grid]]) {</div>
<div class="line"><a id="l00102" name="l00102"></a><span class="lineno"> 102</span> <span class="keyword">auto</span> a_idx = <a class="code hl_function" href="backend_2metal_2kernels_2utils_8h.html#ac612d0ae30b8257198339debe04916a3">elem_to_loc_1&lt;size_t, uint&gt;</a>(index, a_stride);</div>
<div class="line"><a id="l00103" name="l00103"></a><span class="lineno"> 103</span> <span class="keyword">auto</span> b_idx = <a class="code hl_function" href="backend_2metal_2kernels_2utils_8h.html#ac612d0ae30b8257198339debe04916a3">elem_to_loc_1&lt;size_t, uint&gt;</a>(index, b_stride);</div>
<div class="line"><a id="l00102" name="l00102"></a><span class="lineno"> 102</span> <span class="keyword">auto</span> a_idx = <a class="code hl_function" href="backend_2metal_2kernels_2utils_8h.html#ac612d0ae30b8257198339debe04916a3">elem_to_loc_1&lt;size_t, IdxT&gt;</a>(index, a_stride);</div>
<div class="line"><a id="l00103" name="l00103"></a><span class="lineno"> 103</span> <span class="keyword">auto</span> b_idx = <a class="code hl_function" href="backend_2metal_2kernels_2utils_8h.html#ac612d0ae30b8257198339debe04916a3">elem_to_loc_1&lt;size_t, IdxT&gt;</a>(index, b_stride);</div>
<div class="line"><a id="l00104" name="l00104"></a><span class="lineno"> 104</span> <span class="keyword">auto</span> out = Op()(a[a_idx], b[b_idx]);</div>
<div class="line"><a id="l00105" name="l00105"></a><span class="lineno"> 105</span> c[index] = out[0];</div>
<div class="line"><a id="l00106" name="l00106"></a><span class="lineno"> 106</span> d[index] = out[1];</div>
@@ -296,12 +296,12 @@ $(function(){ initResizable(false); });
<div class="ttc" id="ametal_2kernels_2binary__two_8h_html_a12dbda74fa460812177ccb9aeee6e1ca"><div class="ttname"><a href="metal_2kernels_2binary__two_8h.html#a12dbda74fa460812177ccb9aeee6e1ca">binary_vs</a></div><div class="ttdeci">void binary_vs(device const T *a, device const T *b, device U *c, device U *d, uint index)</div><div class="ttdef"><b>Definition</b> binary_two.h:28</div></div>
<div class="ttc" id="ametal_2kernels_2binary__two_8h_html_a12e80730e43dfaa4c79ce8d5f99edc50"><div class="ttname"><a href="metal_2kernels_2binary__two_8h.html#a12e80730e43dfaa4c79ce8d5f99edc50">binary_vv2</a></div><div class="ttdeci">void binary_vv2(device const T *a, device const T *b, device U *c, device U *d, uint2 index, uint2 grid_dim)</div><div class="ttdef"><b>Definition</b> binary_two.h:80</div></div>
<div class="ttc" id="ametal_2kernels_2binary__two_8h_html_a273d2f31691f2c64623c2a97eab344be"><div class="ttname"><a href="metal_2kernels_2binary__two_8h.html#a273d2f31691f2c64623c2a97eab344be">binary_vs2</a></div><div class="ttdeci">void binary_vs2(device const T *a, device const T *b, device U *c, device U *d, uint2 index, uint2 grid_dim)</div><div class="ttdef"><b>Definition</b> binary_two.h:66</div></div>
<div class="ttc" id="ametal_2kernels_2binary__two_8h_html_a44e8241fa805906fa9644204e4d1237f"><div class="ttname"><a href="metal_2kernels_2binary__two_8h.html#a44e8241fa805906fa9644204e4d1237f">binary_g_nd1</a></div><div class="ttdeci">void binary_g_nd1(device const T *a, device const T *b, device U *c, device U *d, constant const size_t &amp;a_stride, constant const size_t &amp;b_stride, uint index)</div><div class="ttdef"><b>Definition</b> binary_two.h:94</div></div>
<div class="ttc" id="ametal_2kernels_2binary__two_8h_html_a97b5613aff654d32c49225209a19bb95"><div class="ttname"><a href="metal_2kernels_2binary__two_8h.html#a97b5613aff654d32c49225209a19bb95">binary_g_nd2</a></div><div class="ttdeci">void binary_g_nd2(device const T *a, device const T *b, device U *c, device U *d, constant const size_t a_strides[2], constant const size_t b_strides[2], uint2 index, uint2 grid_dim)</div><div class="ttdef"><b>Definition</b> binary_two.h:110</div></div>
<div class="ttc" id="ametal_2kernels_2binary__two_8h_html_aae07014f8dffa3649a5c7f4671e1268e"><div class="ttname"><a href="metal_2kernels_2binary__two_8h.html#aae07014f8dffa3649a5c7f4671e1268e">binary_g_nd3</a></div><div class="ttdeci">void binary_g_nd3(device const T *a, device const T *b, device U *c, device U *d, constant const size_t a_strides[3], constant const size_t b_strides[3], uint3 index, uint3 grid_dim)</div><div class="ttdef"><b>Definition</b> binary_two.h:128</div></div>
<div class="ttc" id="ametal_2kernels_2binary__two_8h_html_aaf6edb734cea627bca4f6540dc338fbd"><div class="ttname"><a href="metal_2kernels_2binary__two_8h.html#aaf6edb734cea627bca4f6540dc338fbd">binary_g</a></div><div class="ttdeci">void binary_g(device const T *a, device const T *b, device U *c, device U *d, constant const int *shape, constant const size_t *a_strides, constant const size_t *b_strides, constant const int &amp;ndim, uint3 index, uint3 grid_dim)</div><div class="ttdef"><b>Definition</b> binary_two.h:151</div></div>
<div class="ttc" id="ametal_2kernels_2binary__two_8h_html_ab18c6ecf5065275c93701efd095c916c"><div class="ttname"><a href="metal_2kernels_2binary__two_8h.html#ab18c6ecf5065275c93701efd095c916c">binary_sv</a></div><div class="ttdeci">void binary_sv(device const T *a, device const T *b, device U *c, device U *d, uint index)</div><div class="ttdef"><b>Definition</b> binary_two.h:16</div></div>
<div class="ttc" id="ametal_2kernels_2binary__two_8h_html_ab4324f594c007a6895540b77ad5d89d9"><div class="ttname"><a href="metal_2kernels_2binary__two_8h.html#ab4324f594c007a6895540b77ad5d89d9">binary_vv</a></div><div class="ttdeci">void binary_vv(device const T *a, device const T *b, device U *c, device U *d, uint index)</div><div class="ttdef"><b>Definition</b> binary_two.h:40</div></div>
<div class="ttc" id="ametal_2kernels_2binary__two_8h_html_ad1fad37c168192b212a4294f4cf78133"><div class="ttname"><a href="metal_2kernels_2binary__two_8h.html#ad1fad37c168192b212a4294f4cf78133">binary_g_nd1</a></div><div class="ttdeci">void binary_g_nd1(device const T *a, device const T *b, device U *c, device U *d, constant const size_t &amp;a_stride, constant const size_t &amp;b_stride, uint index)</div><div class="ttdef"><b>Definition</b> binary_two.h:94</div></div>
<div class="ttc" id="ametal_2kernels_2binary__two_8h_html_af8a791ac7ca88d32cd8f4e9ac0f9ab4f"><div class="ttname"><a href="metal_2kernels_2binary__two_8h.html#af8a791ac7ca88d32cd8f4e9ac0f9ab4f">binary_ss</a></div><div class="ttdeci">void binary_ss(device const T *a, device const T *b, device U *c, device U *d, uint index)</div><div class="ttdef"><b>Definition</b> binary_two.h:4</div></div>
</div><!-- fragment --></div><!-- contents -->
<!-- start footer part -->