File tree Expand file tree Collapse file tree 1 file changed +1
-11
lines changed
python/paddle/distributed/fleet/recompute Expand file tree Collapse file tree 1 file changed +1
-11
lines changed Original file line number Diff line number Diff line change @@ -475,17 +475,7 @@ def inner_pack(inner_x):
475
475
storage [holder_list [unpack_counter - 1 ]()] = inner_x
476
476
else :
477
477
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 )
489
479
else :
490
480
tmp_tensor = core .eager .Tensor (
491
481
inner_x .dtype ,
You can’t perform that action at this time.
0 commit comments