This commit is contained in:
Awni Hannun
2024-09-17 12:06:14 -07:00
committed by CircleCI Docs
parent 9da49a07a4
commit d44f06ae79
739 changed files with 28107 additions and 8524 deletions

View File

@@ -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_ -&gt; 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_-&gt;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>