From 4e391fe74180f05d82393e8e689840c04f5921c2 Mon Sep 17 00:00:00 2001 From: lichun Date: Wed, 13 Jan 2021 15:34:44 +0800 Subject: [PATCH] abandon using EXPERIMENTAL_DYNAMIC_PARTITION --- ge/graph/partition/dynamic_shape_partition.cc | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/ge/graph/partition/dynamic_shape_partition.cc b/ge/graph/partition/dynamic_shape_partition.cc index 2ec501a8..71a4b560 100755 --- a/ge/graph/partition/dynamic_shape_partition.cc +++ b/ge/graph/partition/dynamic_shape_partition.cc @@ -48,7 +48,7 @@ namespace ge { using Cluster = DynamicShapePartitioner::Cluster; using ClusterPtr = std::shared_ptr; -static bool IsContainResourceOp(const ComputeGraphPtr &root_graph) { +static bool IsInExperimentalMode(const ComputeGraphPtr &root_graph) { for (const auto &node : root_graph->GetAllNodes()) { GE_CHECK_NOTNULL(node->GetOpDesc()); for (const auto &input_desc : node->GetOpDesc()->GetAllInputsDesc()) { @@ -83,7 +83,7 @@ static bool IsContainResourceOp(const ComputeGraphPtr &root_graph) { Status DynamicShapePartitioner::Partition() { REQUIRE_NOT_NULL(root_graph_, "Graph is nullptr."); - if (!IsContainResourceOp(root_graph_)) { + if (!IsInExperimentalMode(root_graph_)) { GELOGD("Skip dynamic shape partition as not in experimental mode."); REQUIRE(AttrUtils::SetBool(*root_graph_, ATTR_NAME_DYNAMIC_SHAPE_PARTITIONED, false), "Failed set dynamic shape partitioned flag on root graph.");