From ab00d4476262634ea2bb333f1b7baa54e6b6499f Mon Sep 17 00:00:00 2001 From: xuexixi Date: Fri, 31 Oct 2025 14:18:33 +0800 Subject: [PATCH] fix sp norm --- .../hybrid_strategy/semi_auto_parallel_simple_net_sp.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/test/auto_parallel/hybrid_strategy/semi_auto_parallel_simple_net_sp.py b/test/auto_parallel/hybrid_strategy/semi_auto_parallel_simple_net_sp.py index 357b248ce51dcb..c3e63bee7b0c44 100644 --- a/test/auto_parallel/hybrid_strategy/semi_auto_parallel_simple_net_sp.py +++ b/test/auto_parallel/hybrid_strategy/semi_auto_parallel_simple_net_sp.py @@ -121,7 +121,7 @@ def forward(self, x): if self.is_sp: out = dist.reshard(out, self.pp0_mesh, self.sp_reshard_placement0) - # out = out + tgt + out = out + tgt out = self.norm(out) out = dist.reshard(out, self.pp1_mesh, self.sp_reshard_placement1)