Skip to content

GitLab

Commit a6b9d13a authored by valarLip's avatar valarLip
Browse files

remod

parent 27e14b4c
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 4 deletions
example/ck_tile/15_fused_moe/instances/fused_moesorting_api.cpp
View file @ a6b9d13a
...@@ -30,10 +30,10 @@ ...@@ -30,10 +30,10 @@
constexpr bool sub_token_onshot = sub_token_onshot_; \ constexpr bool sub_token_onshot = sub_token_onshot_; \
constexpr bool local_expert_masking = local_expert_masking_; \ constexpr bool local_expert_masking = local_expert_masking_; \
using ms_problem = ck_tile::MoeSortingProblemEx<index_t, \ using ms_problem = ck_tile::MoeSortingProblemEx<index_t, \
ms_weight_type, \ ms_weight_type, \
sub_token_tile, \ sub_token_tile, \
sub_token_onshot, \ sub_token_onshot, \
local_expert_masking>; \ local_expert_masking>; \
using kernel = ck_tile::MoeSortingKernel<ms_problem>; \ using kernel = ck_tile::MoeSortingKernel<ms_problem>; \
auto kargs = kernel::MakeKargs(a); \ auto kargs = kernel::MakeKargs(a); \
const dim3 grids = kernel::GridSize(a); \ const dim3 grids = kernel::GridSize(a); \
......
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