From 20335b7c0b42d0882427d044f2ddfd7b6b562d96 Mon Sep 17 00:00:00 2001 From: YuanRisheng Date: Thu, 13 Oct 2022 14:50:42 +0800 Subject: [PATCH] fix bugs (#46951) --- paddle/fluid/framework/infershape_utils.cc | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/paddle/fluid/framework/infershape_utils.cc b/paddle/fluid/framework/infershape_utils.cc index de9f6a4745f..59355c94204 100644 --- a/paddle/fluid/framework/infershape_utils.cc +++ b/paddle/fluid/framework/infershape_utils.cc @@ -490,9 +490,18 @@ std::vector CompatInferMetaContext::MutableOutputBetween( size_t start, size_t end) { std::vector result; result.reserve(end - start); + bool has_meta_tensor = false; + for (size_t i = start; i < end; ++i) { auto& out = compat_outputs_.at(i); result.emplace_back(out.initialized() ? &out : nullptr); + if (!has_meta_tensor && out.initialized()) { + has_meta_tensor = true; + } + } + + if (!has_meta_tensor) { + result.clear(); } return result; } -- GitLab