From f2ebbce77b557acd4f3124a1363a643ccf5d69fa Mon Sep 17 00:00:00 2001 From: lzydev Date: Fri, 25 Aug 2023 11:35:34 +0800 Subject: [PATCH] fix static_build for pp (#56643) --- .../new_executor/interpreter/static_build.cc | 17 ++++++++++------- 1 file changed, 10 insertions(+), 7 deletions(-) diff --git a/paddle/fluid/framework/new_executor/interpreter/static_build.cc b/paddle/fluid/framework/new_executor/interpreter/static_build.cc index 10f244b535c..ce988db2620 100644 --- a/paddle/fluid/framework/new_executor/interpreter/static_build.cc +++ b/paddle/fluid/framework/new_executor/interpreter/static_build.cc @@ -49,13 +49,16 @@ std::set StaticBuildBlackList = { "sparse_sparse_coo_tensor" /*: to handle sparse output*/}; // TODO(lizhiyu): This operator list is only for pipeline strategy temporarily. -std::set SkipCheckForPipelineTempList = {"c_broadcast", - "c_allreduce_sum", - "c_allgather", - "layer_norm", - "recv_v2", - "reshape2_grad", - "c_identity"}; +std::set SkipCheckForPipelineTempList = { + "c_broadcast", + "c_allreduce_sum", + "c_allgather", + "layer_norm", + "recv_v2", + "reshape2_grad", + "c_identity", + "c_reduce_sum", +}; namespace paddle { namespace framework { namespace interpreter { -- GitLab