Skip to content

Commit 6b778b9

Browse files
authored
fix pp release_output (#54672)
1 parent 8b818d0 commit 6b778b9

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

python/paddle/distributed/fleet/meta_parallel/pipeline_parallel.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -618,6 +618,7 @@ def can_free(t):
618618
return (
619619
t is not None
620620
and isinstance(t, paddle.Tensor)
621+
and t._is_initialized()
621622
and t.inplace_version == 0
622623
)
623624

0 commit comments

Comments
 (0)