Skip to content

Commit bfa294b

Browse files
committed
use inplace multiply in grad_clip
1 parent 5e0d9d7 commit bfa294b

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

python/paddle/nn/clip.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -824,8 +824,8 @@ def async_add_n(var_list):
824824
clip_input = paddle.distributed.reshard(
825825
clip_input, g.process_mesh, clip_input.placements
826826
)
827-
new_grad = paddle.multiply(g, clip_input)
828-
params_and_grads.append((p, new_grad))
827+
g.multiply_(clip_input)
828+
params_and_grads.append((p, g))
829829
else:
830830
params_and_grads.append((p, g))
831831

@@ -1089,8 +1089,8 @@ def async_add_n(var_list):
10891089
),
10901090
)
10911091

1092-
new_grad = paddle.multiply(g, clip_input)
1093-
params_and_grads.append((p, new_grad))
1092+
g.multiply_(clip_input)
1093+
params_and_grads.append((p, g))
10941094
else:
10951095
params_and_grads.append((p, g))
10961096

0 commit comments

Comments
 (0)