From 5ffd4afe78a7d64028c9c5c5087b94a8623d0630 Mon Sep 17 00:00:00 2001
From: sneaxiy <32832641+sneaxiy@users.noreply.github.com>
Date: Tue, 1 Nov 2022 14:32:15 +0800
Subject: [PATCH] add missing scale parameter (#47522)

---
 python/paddle/distributed/fleet/utils/hybrid_parallel_util.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/python/paddle/distributed/fleet/utils/hybrid_parallel_util.py b/python/paddle/distributed/fleet/utils/hybrid_parallel_util.py
index 7e95bfe7f33..7e527eced3f 100644
--- a/python/paddle/distributed/fleet/utils/hybrid_parallel_util.py
+++ b/python/paddle/distributed/fleet/utils/hybrid_parallel_util.py
@@ -197,7 +197,7 @@ def fused_allreduce_gradients_with_group(
         else _apply_collective_grads
     )
     with framework.no_grad():
-        apply_func(parameter_list, group, bucket_size)
+        apply_func(parameter_list, group, bucket_size, scale)
 
 
 def fused_allreduce_gradients(parameter_list, hcg):
-- 
GitLab