ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-07-17 00:58:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

remove use less code

Browse Source
This commit is contained in:
carlushuang
2025-02-24 00:24:58 +08:00
parent 0944107e02
commit fd71d17c4b
1 changed files with 0 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -148,7 +148,6 @@ CK_TILE_HOST constexpr auto moe_sorting_get_smem_row_col(int tokens_, int num_ex
mask_ = mask_ > 0b111 ? 0b111 : mask_; //clamp to 8x at most
mask_ = ~mask_;
//printf("r_unroll_:%d, clz:%d, mask:%x\n", r_unroll_, clz_, mask_); fflush(stdout);
r_for_sub_token = (r_unroll_ & mask_) * sub_unroll;
}
@@ -161,8 +160,6 @@ CK_TILE_HOST constexpr auto moe_sorting_get_smem_row_col(int tokens_, int num_ex
return r_for_sub_token + cumsum_bufs;
}();
// printf("r:%d, c:%d\n", smem_rows, smem_cols);
return ck_tile::make_tuple(smem_rows, smem_cols);
}
@@ -1838,14 +1835,6 @@ struct MoeSortingMultiPhaseKernel_P3
int position = cumsum - i_show;
prev_cumsum = s[0]; // update the last cumsum
// printf("[%d]bid:%d, tid:%d, e_start:%d, e_end:%d, x:%d(%d,%d), cumsum:%d,
// prev_cumsum:%d, position:%d\n", i, static_cast<int>(blockIdx.x),
// static_cast<int>(threadIdx.x),
// e_start, e_end, x, i_topk, i_show,
// cumsum,prev_cumsum,position
// );
// (void) p_sorted_token_ids; (void) p_sorted_weights;
// (void) p_weights;
if(i_show)
{