add missing scale parameter (#47519)

ad251cb5 · sneaxiy · GitHub · fece00d0 · ad251cb5
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

python/paddle/distributed/fleet/utils/hybrid_parallel_util.py ...on/paddle/distributed/fleet/utils/hybrid_parallel_util.py +1 -1

未找到文件。
--- a/python/paddle/distributed/fleet/utils/hybrid_parallel_util.py
+++ b/python/paddle/distributed/fleet/utils/hybrid_parallel_util.py
@@ -193,7 +193,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):