From 8b1787d4886413c669f53f574b77ddb975e64540 Mon Sep 17 00:00:00 2001
From: taoxiangdong <taoxiangdong1@huawei.com>
Date: Fri, 27 Nov 2020 11:18:53 +0800
Subject: [PATCH] update graph builder compile

---
 ge/graph/build/model_builder.cc | 2 +-
 ge/graph/build/model_builder.h  | 2 ++
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/ge/graph/build/model_builder.cc b/ge/graph/build/model_builder.cc
index 73982f49..02c246e9 100755
--- a/ge/graph/build/model_builder.cc
+++ b/ge/graph/build/model_builder.cc
@@ -816,7 +816,7 @@ Status ModelBuilder::CompileSingleOp() {
   return ge::SUCCESS;
 }
 
-Status ModelBuilder::CheckAicpuOp(const OpDescPtr &op_desc,  std::set<std::string> &cpue_check_set,  std::set<std::string> &tf_engine_set) {
+Status ModelBuilder::CheckAicpuOp(const OpDescPtr &op_desc, std::set<std::string> &cpue_check_set, std::set<std::string> &tf_engine_set) {
   GE_CHECK_NOTNULL(op_desc);
   std::string aicpu_optype;
   bool has_attr_check_cpu = AttrUtils.GetStr(op_desc, "needCheckCpu", aicpu_optype);
diff --git a/ge/graph/build/model_builder.h b/ge/graph/build/model_builder.h
index e75521c7..8efd76e9 100644
--- a/ge/graph/build/model_builder.h
+++ b/ge/graph/build/model_builder.h
@@ -83,6 +83,8 @@ class ModelBuilder {
 
   Status CompileSingleOp();
 
+  Status CheckAicpuOp(const OpDescPtr &op_desc, std::set<std::string> &cpue_check_set, std::set<std::string> &tf_engine_set);
+
   uint64_t session_id_;
 
   map<int64_t, size_t> mem_type_to_mem_offset_;