Skip to content

GitLab

Commit d12c3c6f authored by coderfeli's avatar coderfeli
Browse files

fix k!=1

parent 9d5f21bf
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
include/ck_tile/ops/fused_moe/pipeline/fused_moegemm_pipeline_flatmm_uk.hpp
View file @ d12c3c6f
...@@ -209,9 +209,9 @@ struct FusedMoeGemmPipeline_FlatmmUk ...@@ -209,9 +209,9 @@ struct FusedMoeGemmPipeline_FlatmmUk
threadIdx.x % (BlockShape::Block_K0 / kAlignmentA) * kAlignmentA; threadIdx.x % (BlockShape::Block_K0 / kAlignmentA) * kAlignmentA;
}, },
number<row_ids_a.size()>{}); number<row_ids_a.size()>{});
if (expert_first_token >= kargs.num_tokens) if (expert_first_token&0xffffff >= kargs.num_tokens)
return; return;
// printf("tid %d %d\n", blockIdx.x, threadIdx.x); // printf("tid %d %d, first %d\n", blockIdx.x, threadIdx.x,expert_first_token&0xffffff);
// for (int i = 0; i < row_ids_a.size(); i++) { // for (int i = 0; i < row_ids_a.size(); i++) {
// printf("%d bid %d tid %d rowid %d\n", i, blockIdx.x, threadIdx.x, row_ids_a[i]); // printf("%d bid %d tid %d rowid %d\n", i, blockIdx.x, threadIdx.x, row_ids_a[i]);
// } // }
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment