From 6b778b977face8e83e85a95e41a1adab0593847c Mon Sep 17 00:00:00 2001 From: Haohongxiang <86215757+haohongxiang@users.noreply.github.com> Date: Thu, 15 Jun 2023 10:58:10 +0800 Subject: [PATCH] fix pp release_output (#54672) --- .../paddle/distributed/fleet/meta_parallel/pipeline_parallel.py | 1 + 1 file changed, 1 insertion(+) diff --git a/python/paddle/distributed/fleet/meta_parallel/pipeline_parallel.py b/python/paddle/distributed/fleet/meta_parallel/pipeline_parallel.py index 03e526a64ae..82dd534751a 100755 --- a/python/paddle/distributed/fleet/meta_parallel/pipeline_parallel.py +++ b/python/paddle/distributed/fleet/meta_parallel/pipeline_parallel.py @@ -618,6 +618,7 @@ class PipelineParallel(MetaParallelBase): return ( t is not None and isinstance(t, paddle.Tensor) + and t._is_initialized() and t.inplace_version == 0 ) -- GitLab