Revert "add return logic when no input"

This reverts commit e674fb74.

Revert "add return logic when no input"
This reverts commit e674fb74.
1d1a51ed · coderfeli · e674fb74 · 1d1a51ed
Commit 1d1a51ed authored Jan 13, 2025 by coderfeli
Show whitespace changes
Inline Side-by-side

Showing with 0 additions and 3 deletions

include/ck_tile/ops/fused_moe/pipeline/fused_moegemm_pipeline_flatmm_uk.hpp ...s/fused_moe/pipeline/fused_moegemm_pipeline_flatmm_uk.hpp +0 -3

No files found.
--- a/include/ck_tile/ops/fused_moe/pipeline/fused_moegemm_pipeline_flatmm_uk.hpp
+++ b/include/ck_tile/ops/fused_moe/pipeline/fused_moegemm_pipeline_flatmm_uk.hpp
@@ -313,9 +313,6 @@ struct FusedMoeGemmPipeline_FlatmmUk
        auto w_scale      = GetWeightScale(
            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 acc_0  = uk_0_g(a_res,
                            a_coords,