mirror of
https://github.com/ml-explore/mlx.git
synced 2025-09-18 01:50:16 +08:00
docs up
This commit is contained in:

committed by
CircleCI Docs

parent
9da49a07a4
commit
d44f06ae79
@@ -496,7 +496,7 @@ $(function() { codefold.init(0); });
|
||||
<div class="line"><a id="l00394" name="l00394"></a><span class="lineno"> 394</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="l00395" name="l00395"></a><span class="lineno"> 395</span> uint simd_group_id [[simdgroup_index_in_threadgroup]],</div>
|
||||
<div class="line"><a id="l00396" name="l00396"></a><span class="lineno"> 396</span> uint simd_lane_id [[thread_index_in_simdgroup]])</div>
|
||||
<div class="line"><a id="l00397" name="l00397"></a><span class="lineno"> 397</span> : <a class="code hl_variable" href="structmlx_1_1steel_1_1_conv2_d_weight_block_loader.html#a7464ec687323fa79050702952ed9084f">src_ld</a>(params_ -> wt_strides[0]),</div>
|
||||
<div class="line"><a id="l00397" name="l00397"></a><span class="lineno"> 397</span> : <a class="code hl_variable" href="structmlx_1_1steel_1_1_conv2_d_weight_block_loader.html#a7464ec687323fa79050702952ed9084f">src_ld</a>(params_->wt_strides[0]),</div>
|
||||
<div class="line"><a id="l00398" name="l00398"></a><span class="lineno"> 398</span> <a class="code hl_variable" href="structmlx_1_1steel_1_1_conv2_d_weight_block_loader.html#ab1cb2ade639787243e0325dcd3dc0a11">thread_idx</a>(simd_group_id * 32 + simd_lane_id),</div>
|
||||
<div class="line"><a id="l00399" name="l00399"></a><span class="lineno"> 399</span> <a class="code hl_variable" href="structmlx_1_1steel_1_1_conv2_d_weight_block_loader.html#a8c5e74003600132954cb953616e1a026">bi</a>(<a class="code hl_variable" href="structmlx_1_1steel_1_1_conv2_d_weight_block_loader.html#ab1cb2ade639787243e0325dcd3dc0a11">thread_idx</a> / <a class="code hl_variable" href="structmlx_1_1steel_1_1_conv2_d_weight_block_loader.html#a80cb90674f839d5d4ecfde384fa0a7a2">TCOLS</a>),</div>
|
||||
<div class="line"><a id="l00400" name="l00400"></a><span class="lineno"> 400</span> <a class="code hl_variable" href="structmlx_1_1steel_1_1_conv2_d_weight_block_loader.html#a6f2fdcaf5a67567cca38ae3d8120ab37">bj</a>(<a class="code hl_variable" href="structmlx_1_1steel_1_1_conv2_d_weight_block_loader.html#a10109dc9553207f5a365799e4969c6d2">vec_size</a> * (<a class="code hl_variable" href="structmlx_1_1steel_1_1_conv2_d_weight_block_loader.html#ab1cb2ade639787243e0325dcd3dc0a11">thread_idx</a> % <a class="code hl_variable" href="structmlx_1_1steel_1_1_conv2_d_weight_block_loader.html#a80cb90674f839d5d4ecfde384fa0a7a2">TCOLS</a>)),</div>
|
||||
|
Reference in New Issue
Block a user