From 257bc844dcd680f02d2bd85f7a7300de3b132d8e Mon Sep 17 00:00:00 2001
From: wxl <wanxuelei@huawei.com>
Date: Thu, 17 Dec 2020 21:21:14 +0800
Subject: [PATCH] bugfix

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

diff --git a/ge/ir_build/ge_ir_build.cc b/ge/ir_build/ge_ir_build.cc
index 0e1d9452..84d75b27 100644
--- a/ge/ir_build/ge_ir_build.cc
+++ b/ge/ir_build/ge_ir_build.cc
@@ -317,6 +317,13 @@ graphStatus Impl::CheckOptions(const std::map<std::string, std::string> &options
   if (it != options_.end() && (CheckDisableReuseMemoryParamValid(it->second) != GRAPH_SUCCESS)) {
     return GRAPH_PARAM_INVALID;
   }
+  // Check Input Format
+  if (options_.find(kInputFormat) != options_.end()) {
+    if (!options_[kInputFormat].empty() && !ge::TypeUtils::IsFormatValid(options_[kInputFormat].c_str())) {
+      GELOGE(ge::PARAM_INVALID, "user input format [%s] is not found!", options_[kInputFormat].c_str());
+      return GRAPH_PARAM_INVALID;
+    }
+  }
   return GRAPH_SUCCESS;
 }