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

docs up

Browse Source
This commit is contained in:
Awni Hannun
2024-09-17 12:06:14 -07:00
committed by CircleCI Docs
parent 27e07888b0
commit 6a6ffb598d
739 changed files with 28107 additions and 8524 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
docs/build/html/loader__channel__n_8h_source.html vendored
View File

@@ -347,7 +347,7 @@ $(function() { codefold.init(0); });
<div class="line"><a id="l00244" name="l00244"></a><span class="lineno"> 244</span> <span class="keyword">const</span> constant <a class="code hl_struct" href="structmlx_1_1steel_1_1_implicit_gemm_conv2_d_params.html">ImplicitGemmConv2DParams</a>* gemm_params_,</div>
<div class="line"><a id="l00245" name="l00245"></a><span class="lineno"> 245</span> uint simd_group_id [[simdgroup_index_in_threadgroup]],</div>
<div class="line"><a id="l00246" name="l00246"></a><span class="lineno"> 246</span> uint simd_lane_id [[thread_index_in_simdgroup]])</div>
<div class="line"><a id="l00247" name="l00247"></a><span class="lineno"> 247</span> : <a class="code hl_variable" href="structmlx_1_1steel_1_1_conv2_d_weight_block_loader_small_channels.html#aa6bedc0cbb447eaf70c03f2e26df2cb2">src_ld</a>(params_ -&gt; wt_strides[0]),</div>
<div class="line"><a id="l00247" name="l00247"></a><span class="lineno"> 247</span> : <a class="code hl_variable" href="structmlx_1_1steel_1_1_conv2_d_weight_block_loader_small_channels.html#aa6bedc0cbb447eaf70c03f2e26df2cb2">src_ld</a>(params_-&gt;wt_strides[0]),</div>
<div class="line"><a id="l00248" name="l00248"></a><span class="lineno"> 248</span> <a class="code hl_variable" href="structmlx_1_1steel_1_1_conv2_d_weight_block_loader_small_channels.html#acacdac168004c87fee27c8554ac905a7">thread_idx</a>(simd_group_id * 32 + simd_lane_id),</div>
<div class="line"><a id="l00249" name="l00249"></a><span class="lineno"> 249</span> <a class="code hl_variable" href="structmlx_1_1steel_1_1_conv2_d_weight_block_loader_small_channels.html#ae3b9f21f72e5e6c541c9978f55d354c7">bi</a>(<a class="code hl_variable" href="structmlx_1_1steel_1_1_conv2_d_weight_block_loader_small_channels.html#acacdac168004c87fee27c8554ac905a7">thread_idx</a> / <a class="code hl_variable" href="structmlx_1_1steel_1_1_conv2_d_weight_block_loader_small_channels.html#acd54132d0928d0f6fb15b2f367e5d5e8">TCOLS</a>),</div>
<div class="line"><a id="l00250" name="l00250"></a><span class="lineno"> 250</span> <a class="code hl_variable" href="structmlx_1_1steel_1_1_conv2_d_weight_block_loader_small_channels.html#adaa261fc2e8e694aedab4ebd60b52e5e">bj</a>(<a class="code hl_variable" href="structmlx_1_1steel_1_1_conv2_d_weight_block_loader_small_channels.html#a006153d274aa13d5fd4448b4607fff3a">vec_size</a> * (<a class="code hl_variable" href="structmlx_1_1steel_1_1_conv2_d_weight_block_loader_small_channels.html#acacdac168004c87fee27c8554ac905a7">thread_idx</a> % <a class="code hl_variable" href="structmlx_1_1steel_1_1_conv2_d_weight_block_loader_small_channels.html#acd54132d0928d0f6fb15b2f367e5d5e8">TCOLS</a>)),</div>