Skip to content

Commit 5ffd4af

Browse files
authored
add missing scale parameter (#47522)
1 parent 0201ccc commit 5ffd4af

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

python/paddle/distributed/fleet/utils/hybrid_parallel_util.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -197,7 +197,7 @@ def fused_allreduce_gradients_with_group(
197197
else _apply_collective_grads
198198
)
199199
with framework.no_grad():
200-
apply_func(parameter_list, group, bucket_size)
200+
apply_func(parameter_list, group, bucket_size, scale)
201201

202202

203203
def fused_allreduce_gradients(parameter_list, hcg):

0 commit comments

Comments
 (0)