ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-06-29 19:28:33 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix typo

Browse Source
This commit is contained in:
coderfeli
2025-09-09 08:15:38 +00:00
parent e57a8b3521
commit 368e87ad36
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
include/ck_tile/ops/fused_moe/kernel/moe_sorting_kernel.hpp
View File

@@ -1713,7 +1713,7 @@ struct MoeSortingMultiPhaseKernel_P0
int loops = (mesh_stride / index_pack + BLOCK_SIZE - 1) / BLOCK_SIZE;
if(problem::LocalToken && mask == 0)
if(Problem::LocalToken && mask == 0)
return; // skip
index_t cnt = 0; // per-wave cnt
for(int i = 0; i < loops; i++)
@@ -1721,7 +1721,7 @@ struct MoeSortingMultiPhaseKernel_P0
int position = i * BLOCK_SIZE + threadIdx.x;
r_t v{0};
if(position < (mesh_stride / index_pack))
v = p_expert_mesh[position];
v = p_expert_mesh_r[position];
index_t local_sum = 0;
static_for<0, index_pack, 1>{}(
[&](auto i_vec) { local_sum += v[i_vec.value] != 0 ? 1 : 0; });