Commit e674fb74 authored by coderfeli's avatar coderfeli
Browse files

add return logic when no input

parent 59c05300
...@@ -313,6 +313,9 @@ struct FusedMoeGemmPipeline_FlatmmUk ...@@ -313,6 +313,9 @@ struct FusedMoeGemmPipeline_FlatmmUk
auto w_scale = GetWeightScale( auto w_scale = GetWeightScale(
row_coords_o, reinterpret_cast<const TopkWeightDataType*>(kargs.sorted_weight_ptr)); row_coords_o, reinterpret_cast<const TopkWeightDataType*>(kargs.sorted_weight_ptr));
if (row_ids_a[0] >= kargs.num_tokens)
return;
auto uk_0_g = Policy::template GetUK_0<Problem>(); auto uk_0_g = Policy::template GetUK_0<Problem>();
auto acc_0 = uk_0_g(a_res, auto acc_0 = uk_0_g(a_res,
a_coords, a_coords,
......
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