From c60279ac6284f1c7fc0ade65a100298290e19860 Mon Sep 17 00:00:00 2001
From: wxl <wanxuelei@huawei.com>
Date: Thu, 10 Dec 2020 14:05:48 +0800
Subject: [PATCH] ir build optimize

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

diff --git a/ge/ir_build/ge_ir_build.cc b/ge/ir_build/ge_ir_build.cc
index b353a72a..f9c4e259 100644
--- a/ge/ir_build/ge_ir_build.cc
+++ b/ge/ir_build/ge_ir_build.cc
@@ -266,6 +266,8 @@ graphStatus Impl::UpdateDataOpAttr(const Graph &graph) {
     if (op->GetType() == DATA) {
       auto tensor_input = op->MutableInputDesc(0);
       auto tensor_output = op->MutableOutputDesc(0);
+      GE_CHECK_NOTNULL(tensor_input);
+      GE_CHECK_NOTNULL(tensor_output);
       string data_op_name = op->GetName();
       auto iter = shape_map.find(data_op_name);
       if (iter != shape_map.end()) {
@@ -317,8 +319,7 @@ graphStatus Impl::CheckOptions(const std::map<std::string, std::string> &options
   }
   // Check option EXEC_DISABLE_REUSED_MEMORY
   it = options_.find(ge::ir_option::EXEC_DISABLE_REUSED_MEMORY);
-  if (it != options_.end() && it->second != kReUseMemEnable && it->second != kReUseMemDisEnable) {
-    GELOGE(GRAPH_PARAM_INVALID, "option(EXEC_DISABLE_REUSED_MEMORY) value[%s] is invalid ", it->second.c_str());
+  if (it != options_.end() && (CheckDisableReuseMemoryParamValid(it->second) != GRAPH_SUCCESS)) {
     return GRAPH_PARAM_INVALID;
   }
   return GRAPH_SUCCESS;