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

docs update

Browse Source
This commit is contained in:
Awni Hannun
2024-08-23 12:14:53 -07:00
committed by CircleCI Docs
parent d8f7b8cda6
commit 27e07888b0
697 changed files with 15867 additions and 8594 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
docs/build/html/kernels_2softmax_8h_source.html vendored
View File

@@ -208,7 +208,7 @@ $(function() { codefold.init(0); });
<div class="line"><a id="l00118" name="l00118"></a><span class="lineno"> 118</span> AccT prevmax;</div>
<div class="line"><a id="l00119" name="l00119"></a><span class="lineno"> 119</span> AccT maxval = <a class="code hl_struct" href="struct_limits.html">Limits&lt;AccT&gt;::finite_min</a>;</div>
<div class="line"><a id="l00120" name="l00120"></a><span class="lineno"> 120</span> AccT normalizer = 0;</div>
<div class="line"><a id="l00121" name="l00121"></a><span class="lineno"> 121</span> <span class="keywordflow">for</span> (<span class="keywordtype">int</span> r = 0; r &lt; static_cast&lt;int&gt;(<a class="code hl_function" href="backend_2metal_2kernels_2utils_8h.html#a51c19db777f43943e4b35f25dd88d49d">ceildiv</a>(axis_size, N_READS * lsize));</div>
<div class="line"><a id="l00121" name="l00121"></a><span class="lineno"> 121</span> <span class="keywordflow">for</span> (<span class="keywordtype">int</span> r = 0; r &lt; static_cast&lt;int&gt;(<a class="code hl_function" href="backend_2metal_2kernels_2utils_8h.html#a8e5a4b0fb5d018d7b078d147efe4f1e3">ceildiv</a>(axis_size, N_READS * lsize));</div>
<div class="line"><a id="l00122" name="l00122"></a><span class="lineno"> 122</span> r++) {</div>
<div class="line"><a id="l00123" name="l00123"></a><span class="lineno"> 123</span> <span class="keywordtype">int</span> offset = r * lsize * N_READS + lid * N_READS;</div>
<div class="line"><a id="l00124" name="l00124"></a><span class="lineno"> 124</span> AccT vals[N_READS];</div>
@@ -261,7 +261,7 @@ $(function() { codefold.init(0); });
<div class="line"><a id="l00171" name="l00171"></a><span class="lineno"> 171</span> <span class="comment">// Finally given the normalizer and max value we can directly write the</span></div>
<div class="line"><a id="l00172" name="l00172"></a><span class="lineno"> 172</span> <span class="comment">// softmax output</span></div>
<div class="line"><a id="l00173" name="l00173"></a><span class="lineno"> 173</span> out += gid * size_t(axis_size);</div>
<div class="line"><a id="l00174" name="l00174"></a><span class="lineno"> 174</span> <span class="keywordflow">for</span> (<span class="keywordtype">int</span> r = 0; r &lt; static_cast&lt;int&gt;(<a class="code hl_function" href="backend_2metal_2kernels_2utils_8h.html#a51c19db777f43943e4b35f25dd88d49d">ceildiv</a>(axis_size, N_READS * lsize));</div>
<div class="line"><a id="l00174" name="l00174"></a><span class="lineno"> 174</span> <span class="keywordflow">for</span> (<span class="keywordtype">int</span> r = 0; r &lt; static_cast&lt;int&gt;(<a class="code hl_function" href="backend_2metal_2kernels_2utils_8h.html#a8e5a4b0fb5d018d7b078d147efe4f1e3">ceildiv</a>(axis_size, N_READS * lsize));</div>
<div class="line"><a id="l00175" name="l00175"></a><span class="lineno"> 175</span> r++) {</div>
<div class="line"><a id="l00176" name="l00176"></a><span class="lineno"> 176</span> <span class="keywordtype">int</span> offset = r * lsize * N_READS + lid * N_READS;</div>
<div class="line"><a id="l00177" name="l00177"></a><span class="lineno"> 177</span> <span class="keywordflow">if</span> (offset + N_READS &lt;= axis_size) {</div>
@@ -279,7 +279,7 @@ $(function() { codefold.init(0); });
<div class="line"><a id="l00189" name="l00189"></a><span class="lineno"> 189</span> }</div>
<div class="line"><a id="l00190" name="l00190"></a><span class="lineno"> 190</span>}</div>
</div>
<div class="ttc" id="abackend_2metal_2kernels_2utils_8h_html_a51c19db777f43943e4b35f25dd88d49d"><div class="ttname"><a href="backend_2metal_2kernels_2utils_8h.html#a51c19db777f43943e4b35f25dd88d49d">ceildiv</a></div><div class="ttdeci">size_t ceildiv(size_t N, size_t M)</div><div class="ttdoc">Compute ceil((float)N/(float)M)</div><div class="ttdef"><b>Definition</b> utils.h:296</div></div>
<div class="ttc" id="abackend_2metal_2kernels_2utils_8h_html_a8e5a4b0fb5d018d7b078d147efe4f1e3"><div class="ttname"><a href="backend_2metal_2kernels_2utils_8h.html#a8e5a4b0fb5d018d7b078d147efe4f1e3">ceildiv</a></div><div class="ttdeci">T ceildiv(T N, U M)</div><div class="ttdoc">Compute ceil((float)N/(float)M)</div><div class="ttdef"><b>Definition</b> utils.h:409</div></div>
<div class="ttc" id="akernels_2softmax_8h_html_a440d4031ee5e86159a4dd715e44a438b"><div class="ttname"><a href="kernels_2softmax_8h.html#a440d4031ee5e86159a4dd715e44a438b">softmax_exp</a></div><div class="ttdeci">T softmax_exp(T x)</div><div class="ttdef"><b>Definition</b> softmax.h:4</div></div>
<div class="ttc" id="akernels_2softmax_8h_html_a815fe70f879f318e5d6e99acf043f52b"><div class="ttname"><a href="kernels_2softmax_8h.html#a815fe70f879f318e5d6e99acf043f52b">softmax_single_row</a></div><div class="ttdeci">void softmax_single_row(const device T *in, device T *out, constant int &amp;axis_size, uint gid, uint _lid, uint simd_lane_id, uint simd_group_id)</div><div class="ttdef"><b>Definition</b> softmax.h:11</div></div>
<div class="ttc" id="akernels_2softmax_8h_html_a8c47b0924ebfeebcca25f3dd17373276"><div class="ttname"><a href="kernels_2softmax_8h.html#a8c47b0924ebfeebcca25f3dd17373276">softmax_looped</a></div><div class="ttdeci">void softmax_looped(const device T *in, device T *out, constant int &amp;axis_size, uint gid, uint lid, uint lsize, uint simd_lane_id, uint simd_group_id)</div><div class="ttdef"><b>Definition</b> softmax.h:101</div></div>