From 939fb61fa1c3833895a94a02d36402af5d296f56 Mon Sep 17 00:00:00 2001 From: zhaozhixuan Date: Sat, 15 May 2021 16:16:13 +0800 Subject: [PATCH] Add shape range check. --- ge/graph/manager/graph_manager.cc | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/ge/graph/manager/graph_manager.cc b/ge/graph/manager/graph_manager.cc index 6e8478d8..02a500fe 100755 --- a/ge/graph/manager/graph_manager.cc +++ b/ge/graph/manager/graph_manager.cc @@ -490,8 +490,8 @@ Status GraphManager::ModifyDataIndex(const Graph &graph, const std::mapsecond.empty()); - if (enable_input_shape_range) { + if (iter != graph_option.end()) && (!iter->second.empty()) { + // If data inputs shape range is set, user must set valid data index. REPORT_INNER_ERROR("E19999", "Input data index is invalid when data shape range enabled, please check!"); GELOGE(GRAPH_PARAM_INVALID, "[COMP][AddGraph]Input data index is invalid when data shape range enabled."); return GRAPH_PARAM_INVALID;