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

[BugFix]Dev fix custom ar unstable result (#4437)

Browse Source
This commit is contained in:
chen
2025-10-17 11:47:16 +08:00
committed by GitHub
parent 6160145f82
commit b134e6afe6
17 changed files with 25 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/distributed/custom_all_reduce.py
+1 -1
View File
@@ -57,7 +57,7 @@ class Test(unittest.TestCase):
data_custom_ar = paddle.rand([m, n], dtype="bfloat16")
data_paddle = data_custom_ar.clone()
if fa.should_custom_ar(data_custom_ar):
fa.custom_all_reduce(data_custom_ar)
data_custom_ar = fa.custom_all_reduce(data_custom_ar)
dist.all_reduce(data_paddle)
if dist.get_rank() == 0:
np.testing.assert_allclose(