Skip to content

Commit 1692f42

Browse files
committed
fix recompute
1 parent ce7f373 commit 1692f42

File tree

1 file changed

+1
-11
lines changed

1 file changed

+1
-11
lines changed

python/paddle/distributed/fleet/recompute/recompute.py

Lines changed: 1 addition & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -475,17 +475,7 @@ def inner_pack(inner_x):
475475
storage[holder_list[unpack_counter - 1]()] = inner_x
476476
else:
477477
if inner_x.is_dist():
478-
# TODO(jeff41404): it seems better to use `tmp_tensor = core.eager.Tensor(inner_x)`,
479-
# but other errors will be triggered during the current period, and can be modified after resolution
480-
tmp_tensor = core.eager.Tensor(
481-
inner_x.dtype,
482-
inner_x.shape,
483-
inner_x.name + "cpy",
484-
core.VarDesc.VarType.DENSE_TENSOR,
485-
inner_x.persistable,
486-
inner_x.process_mesh,
487-
inner_x.placements,
488-
)
478+
tmp_tensor = core.eager.Tensor(inner_x)
489479
else:
490480
tmp_tensor = core.eager.Tensor(
491481
inner_x.dtype,

0 commit comments

Comments
 (0)