apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2026-04-23 08:21:53 +08:00
Code Issues Actions 19 Packages Projects Releases Wiki Activity

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

Browse Source 
This reverts commit 2bb838fed9.
This commit is contained in:
Yuanle Liu
2026-01-09 15:20:40 +08:00
committed by GitHub
parent 1d20957340
commit 8c3513a410
9 changed files with 8 additions and 30 deletions
Download Patch File Download Diff File Expand all files Collapse all files
fastdeploy/model_executor/models/ernie4_5_moe.py
-3
View File
@@ -477,9 +477,6 @@ class Ernie4_5_Model(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])
if current_platform.is_iluvatar() and forward_meta.attn_backend.mixed:
out = forward_meta.attn_backend.reverse_transpose(out)