From dfa45903733f51aaad59dd453406209edd26d925 Mon Sep 17 00:00:00 2001
From: l00444296 <lixiwen1@huawei.com>
Date: Tue, 8 Dec 2020 14:26:01 +0800
Subject: [PATCH] Feature: Get default from ge ir graph while no user input
 shape

---
 ge/ir_build/ge_ir_build.cc | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/ge/ir_build/ge_ir_build.cc b/ge/ir_build/ge_ir_build.cc
index b0f1b25e..fbacf527 100644
--- a/ge/ir_build/ge_ir_build.cc
+++ b/ge/ir_build/ge_ir_build.cc
@@ -334,7 +334,8 @@ graphStatus Impl::Init(const Graph &graph, const std::map<std::string, std::stri
 
   string input_shape;
   if (options_.find("input_shape") == options_.end()) {
-    GE_CHK_BOOL_RET_STATUS_NOLOG(GetDefaultInputShape(graph, input_shape), GRAPH_PARAM_INVALID);
+    GE_CHK_BOOL_EXEC(GetDefaultInputShape(graph, input_shape) == ge::SUCCESS, return ge::GRAPH_PARAM_INVALID,
+                     "Get defaule data op shape from graph failed!");
   } else {
     input_shape = options_["input_shape"];
   }