From 80795addb1155350fee77a89eb06d7abe99b3e8f Mon Sep 17 00:00:00 2001
From: taoxiangdong <taoxiangdong1@huawei.com>
Date: Sat, 12 Dec 2020 20:04:19 +0800
Subject: [PATCH] update parameter

---
 ge/graph/build/model_builder.cc | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/ge/graph/build/model_builder.cc b/ge/graph/build/model_builder.cc
index b68b5f60..b451b897 100755
--- a/ge/graph/build/model_builder.cc
+++ b/ge/graph/build/model_builder.cc
@@ -609,7 +609,7 @@ Status ModelBuilder::SaveDataToModel(ge::Model &model, ge::GeModel &ge_model) {
     tbe_kernel_store_.AddTBEKernel(tbe_kernel);
   }
 
-  (void)SetModelCheckAicpuAttr(model);
+  (void)SetModelCheckAicpuAttr(model, aicpu_optype_set, aicpu_tf_optype_set);
 
   for (const ge::NodePtr &n : compute_graph_->GetNodes(compute_graph_->GetGraphUnknownFlag())) {
     auto node_op_desc = n->GetOpDesc();