add missing scale parameter (#47522)

PaddlePaddle · Nov 1, 2022 · 5ffd4af · 5ffd4af
1 parent 0201ccc
commit 5ffd4af
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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):