add missing scale parameter (#47522)

5ffd4afe · sneaxiy · GitHub · 0201ccc4 · 5ffd4afe
隐藏空白更改
内联并排

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
@@ -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):