diff --git a/example/ck_tile/13_moe_sorting/moe_sorting_api.cpp b/example/ck_tile/13_moe_sorting/moe_sorting_api.cpp index 1c38b2f1ee..d69adbcb92 100644 --- a/example/ck_tile/13_moe_sorting/moe_sorting_api.cpp +++ b/example/ck_tile/13_moe_sorting/moe_sorting_api.cpp @@ -311,7 +311,6 @@ float moe_sorting(moe_sorting_trait t, moe_sorting_args a, ck_tile::stream_confi { \ float ave_time = \ ck_tile::launch_kernel(s, \ - maybe_clear_workspace, \ MOE_SORTING_MP_0_V2(mesh_type_, token_vec_0_, true, true), \ MOE_SORTING_MP_23(mesh_type_, token_vec_23_, true, true)); \ return ave_time; \ @@ -320,7 +319,6 @@ float moe_sorting(moe_sorting_trait t, moe_sorting_args a, ck_tile::stream_confi { \ float ave_time = \ ck_tile::launch_kernel(s, \ - maybe_clear_workspace, \ MOE_SORTING_MP_0_V2(mesh_type_, token_vec_0_, true, false), \ MOE_SORTING_MP_23(mesh_type_, token_vec_23_, true, false)); \ return ave_time; \ @@ -332,7 +330,6 @@ float moe_sorting(moe_sorting_trait t, moe_sorting_args a, ck_tile::stream_confi { \ float ave_time = \ ck_tile::launch_kernel(s, \ - maybe_clear_workspace, \ MOE_SORTING_MP_0_V2(mesh_type_, token_vec_0_, false, true), \ MOE_SORTING_MP_23(mesh_type_, token_vec_23_, false, true)); \ return ave_time; \ @@ -341,7 +338,6 @@ float moe_sorting(moe_sorting_trait t, moe_sorting_args a, ck_tile::stream_confi { \ float ave_time = ck_tile::launch_kernel( \ s, \ - maybe_clear_workspace, \ MOE_SORTING_MP_0_V2(mesh_type_, token_vec_0_, false, false), \ MOE_SORTING_MP_23(mesh_type_, token_vec_23_, false, false)); \ return ave_time; \ @@ -440,7 +436,6 @@ float moe_sorting_mp(moe_sorting_trait t, moe_sorting_args a, ck_tile::stream_co if(t.local_expert_masking) { float ave_time = ck_tile::launch_kernel(s, - maybe_clear_workspace, MOE_SORTING_MP_0_V2(ms_index_t, 1, true), MOE_SORTING_MP_2(ms_index_t, 1, true), MOE_SORTING_MP_3(ms_index_t, 1, true)); @@ -449,7 +444,6 @@ float moe_sorting_mp(moe_sorting_trait t, moe_sorting_args a, ck_tile::stream_co else { float ave_time = ck_tile::launch_kernel(s, - maybe_clear_workspace, MOE_SORTING_MP_0_V2(ms_index_t, 1, false), MOE_SORTING_MP_2(ms_index_t, 1, false), MOE_SORTING_MP_3(ms_index_t, 1, false));