Skip to content

Commit ead676b

Browse files
authored
[AutoParallel] fix sp test accuracy for H20 (#76150)
1 parent b45fe33 commit ead676b

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

test/auto_parallel/hybrid_strategy/semi_auto_parallel_simple_net_sp.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -121,7 +121,7 @@ def forward(self, x):
121121
if self.is_sp:
122122
out = dist.reshard(out, self.pp0_mesh, self.sp_reshard_placement0)
123123

124-
# out = out + tgt
124+
out = out + tgt
125125
out = self.norm(out)
126126

127127
out = dist.reshard(out, self.pp1_mesh, self.sp_reshard_placement1)

0 commit comments

Comments
 (0)