fix pp release_output (#54672)

PaddlePaddle · Jun 15, 2023 · 6b778b9 · 6b778b9
1 parent 8b818d0
commit 6b778b9
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/python/paddle/distributed/fleet/meta_parallel/pipeline_parallel.py b/python/paddle/distributed/fleet/meta_parallel/pipeline_parallel.py
@@ -618,6 +618,7 @@ def can_free(t):
             return (
                 t is not None
                 and isinstance(t, paddle.Tensor)
+                and t._is_initialized()
                 and t.inplace_version == 0
             )