Browse Source

update graph builder compile

pull/439/head
taoxiangdong 4 years ago
parent
commit
8b1787d488
2 changed files with 3 additions and 1 deletions
  1. +1
    -1
      ge/graph/build/model_builder.cc
  2. +2
    -0
      ge/graph/build/model_builder.h

+ 1
- 1
ge/graph/build/model_builder.cc View File

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


+ 2
- 0
ge/graph/build/model_builder.h View File

@@ -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_;


Loading…
Cancel
Save