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();