zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-09-19 19:38:16 +08:00
Code Issues Actions 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

12
docs/build/html/metal_2kernels_2ternary_8h_source.html vendored
View File

@@ -119,9 +119,9 @@ $(function(){ initResizable(false); });
<div class="line"><a id="l00023" name="l00023"></a><span class="lineno"> 23</span>}</div>
</div>
<div class="line"><a id="l00024" name="l00024"></a><span class="lineno"> 24</span> </div>
<div class="line"><a id="l00025" name="l00025"></a><span class="lineno"> 25</span><span class="keyword">template</span> &lt;<span class="keyword">typename</span> T, <span class="keyword">typename</span> Op&gt;</div>
<div class="line"><a id="l00025" name="l00025"></a><span class="lineno"> 25</span><span class="keyword">template</span> &lt;<span class="keyword">typename</span> T, <span class="keyword">typename</span> Op, <span class="keyword">typename</span> IdxT = <span class="keywordtype">size_t</span>&gt;</div>
<div class="foldopen" id="foldopen00026" data-start="{" data-end="}">
<div class="line"><a id="l00026" name="l00026"></a><span class="lineno"><a class="line" href="metal_2kernels_2ternary_8h.html#a1bd5918559850f3f80e3adee2391fe6a"> 26</a></span>[[kernel]] <span class="keywordtype">void</span> <a class="code hl_function" href="metal_2kernels_2ternary_8h.html#a1bd5918559850f3f80e3adee2391fe6a">ternary_g_nd1</a>(</div>
<div class="line"><a id="l00026" name="l00026"></a><span class="lineno"><a class="line" href="metal_2kernels_2ternary_8h.html#ad63096b65467d3048eefa68f4109878d"> 26</a></span>[[kernel]] <span class="keywordtype">void</span> <a class="code hl_function" href="metal_2kernels_2ternary_8h.html#ad63096b65467d3048eefa68f4109878d">ternary_g_nd1</a>(</div>
<div class="line"><a id="l00027" name="l00027"></a><span class="lineno"> 27</span> device <span class="keyword">const</span> <span class="keywordtype">bool</span>* a,</div>
<div class="line"><a id="l00028" name="l00028"></a><span class="lineno"> 28</span> device <span class="keyword">const</span> T* b,</div>
<div class="line"><a id="l00029" name="l00029"></a><span class="lineno"> 29</span> device <span class="keyword">const</span> T* c,</div>
@@ -130,9 +130,9 @@ $(function(){ initResizable(false); });
<div class="line"><a id="l00032" name="l00032"></a><span class="lineno"> 32</span> constant <span class="keyword">const</span> <span class="keywordtype">size_t</span>&amp; b_strides,</div>
<div class="line"><a id="l00033" name="l00033"></a><span class="lineno"> 33</span> constant <span class="keyword">const</span> <span class="keywordtype">size_t</span>&amp; c_strides,</div>
<div class="line"><a id="l00034" name="l00034"></a><span class="lineno"> 34</span> uint index [[thread_position_in_grid]]) {</div>
<div class="line"><a id="l00035" name="l00035"></a><span class="lineno"> 35</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_strides);</div>
<div class="line"><a id="l00036" name="l00036"></a><span class="lineno"> 36</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_strides);</div>
<div class="line"><a id="l00037" name="l00037"></a><span class="lineno"> 37</span> <span class="keyword">auto</span> c_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, c_strides);</div>
<div class="line"><a id="l00035" name="l00035"></a><span class="lineno"> 35</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_strides);</div>
<div class="line"><a id="l00036" name="l00036"></a><span class="lineno"> 36</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_strides);</div>
<div class="line"><a id="l00037" name="l00037"></a><span class="lineno"> 37</span> <span class="keyword">auto</span> c_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, c_strides);</div>
<div class="line"><a id="l00038" name="l00038"></a><span class="lineno"> 38</span> d[index] = Op()(a[a_idx], b[b_idx], c[c_idx]);</div>
<div class="line"><a id="l00039" name="l00039"></a><span class="lineno"> 39</span>}</div>
</div>
@@ -216,10 +216,10 @@ $(function(){ initResizable(false); });
<div class="ttc" id="abackend_2metal_2kernels_2utils_8h_html_a65d87b425e1f8ca19df97c15049f8733"><div class="ttname"><a href="backend_2metal_2kernels_2utils_8h.html#a65d87b425e1f8ca19df97c15049f8733">elem_to_loc_3_nd</a></div><div class="ttdeci">METAL_FUNC vec&lt; IdxT, 3 &gt; elem_to_loc_3_nd(uint3 elem, constant const int *shape, constant const size_t *a_strides, constant const size_t *b_strides, constant const size_t *c_strides, int ndim)</div><div class="ttdef"><b>Definition</b> utils.h:182</div></div>
<div class="ttc" id="abackend_2metal_2kernels_2utils_8h_html_ac612d0ae30b8257198339debe04916a3"><div class="ttname"><a href="backend_2metal_2kernels_2utils_8h.html#ac612d0ae30b8257198339debe04916a3">elem_to_loc_1</a></div><div class="ttdeci">METAL_FUNC IdxT elem_to_loc_1(uint elem, constant const StrideT &amp;stride)</div><div class="ttdef"><b>Definition</b> utils.h:140</div></div>
<div class="ttc" id="ametal_2kernels_2ternary_8h_html_a046dcbf67cd2318d45355dc7516e3ff4"><div class="ttname"><a href="metal_2kernels_2ternary_8h.html#a046dcbf67cd2318d45355dc7516e3ff4">ternary_g_nd3</a></div><div class="ttdeci">void ternary_g_nd3(device const bool *a, device const T *b, device const T *c, device T *d, constant const size_t a_strides[3], constant const size_t b_strides[3], constant const size_t c_strides[3], uint3 index, uint3 grid_dim)</div><div class="ttdef"><b>Definition</b> ternary.h:60</div></div>
<div class="ttc" id="ametal_2kernels_2ternary_8h_html_a1bd5918559850f3f80e3adee2391fe6a"><div class="ttname"><a href="metal_2kernels_2ternary_8h.html#a1bd5918559850f3f80e3adee2391fe6a">ternary_g_nd1</a></div><div class="ttdeci">void ternary_g_nd1(device const bool *a, device const T *b, device const T *c, device T *d, constant const size_t &amp;a_strides, constant const size_t &amp;b_strides, constant const size_t &amp;c_strides, uint index)</div><div class="ttdef"><b>Definition</b> ternary.h:26</div></div>
<div class="ttc" id="ametal_2kernels_2ternary_8h_html_a3e610f3b01966bdbf23fdfebe5d2c508"><div class="ttname"><a href="metal_2kernels_2ternary_8h.html#a3e610f3b01966bdbf23fdfebe5d2c508">ternary_v2</a></div><div class="ttdeci">void ternary_v2(device const bool *a, device const T *b, device const T *c, device T *d, uint2 index, uint2 grid_dim)</div><div class="ttdef"><b>Definition</b> ternary.h:14</div></div>
<div class="ttc" id="ametal_2kernels_2ternary_8h_html_a83f93644d21ee774e06e8190d0725ccb"><div class="ttname"><a href="metal_2kernels_2ternary_8h.html#a83f93644d21ee774e06e8190d0725ccb">ternary_v</a></div><div class="ttdeci">void ternary_v(device const bool *a, device const T *b, device const T *c, device T *d, uint index)</div><div class="ttdef"><b>Definition</b> ternary.h:4</div></div>
<div class="ttc" id="ametal_2kernels_2ternary_8h_html_ab2051fd944c2e24c57d5b4af54894d72"><div class="ttname"><a href="metal_2kernels_2ternary_8h.html#ab2051fd944c2e24c57d5b4af54894d72">ternary_g</a></div><div class="ttdeci">void ternary_g(device const bool *a, device const T *b, device const T *c, device T *d, constant const int *shape, constant const size_t *a_strides, constant const size_t *b_strides, constant const size_t *c_strides, constant const int &amp;ndim, uint3 index, uint3 grid_dim)</div><div class="ttdef"><b>Definition</b> ternary.h:78</div></div>
<div class="ttc" id="ametal_2kernels_2ternary_8h_html_ad63096b65467d3048eefa68f4109878d"><div class="ttname"><a href="metal_2kernels_2ternary_8h.html#ad63096b65467d3048eefa68f4109878d">ternary_g_nd1</a></div><div class="ttdeci">void ternary_g_nd1(device const bool *a, device const T *b, device const T *c, device T *d, constant const size_t &amp;a_strides, constant const size_t &amp;b_strides, constant const size_t &amp;c_strides, uint index)</div><div class="ttdef"><b>Definition</b> ternary.h:26</div></div>
<div class="ttc" id="ametal_2kernels_2ternary_8h_html_adec9ca8a8bf527cb15d70da5857af15d"><div class="ttname"><a href="metal_2kernels_2ternary_8h.html#adec9ca8a8bf527cb15d70da5857af15d">ternary_g_nd2</a></div><div class="ttdeci">void ternary_g_nd2(device const bool *a, device const T *b, device const T *c, device T *d, constant const size_t a_strides[2], constant const size_t b_strides[2], constant const size_t c_strides[2], uint2 index, uint2 grid_dim)</div><div class="ttdef"><b>Definition</b> ternary.h:42</div></div>
</div><!-- fragment --></div><!-- contents -->
<!-- start footer part -->