Revert "[TSP] last_norm allgather move to model.py (#5924)" (#5961)

This reverts commit 2bb838fed9.
2026-04-23 17:11:21 +08:00 · 2026-01-09 15:20:40 +08:00
parent 1d20957340
commit 8c3513a410
9 changed files with 8 additions and 30 deletions
@@ -592,9 +592,6 @@ class DeepSeekV3Model(nn.Layer):
            )
        out = self.norm(hidden_states, residual, forward_meta=forward_meta)[0]

-        if self.norm.is_last_norm and self.norm.fd_config.parallel_config.use_sequence_parallel_moe:
-            out = self.norm.allgather(out, forward_meta.ids_remove_padding.shape[0])
-
        return out