Browse Source

update graph builder compile

pull/439/head
taoxiangdong 5 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; 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); GE_CHECK_NOTNULL(op_desc);
std::string aicpu_optype; std::string aicpu_optype;
bool has_attr_check_cpu = AttrUtils.GetStr(op_desc, "needCheckCpu", 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 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_; uint64_t session_id_;


map<int64_t, size_t> mem_type_to_mem_offset_; map<int64_t, size_t> mem_type_to_mem_offset_;


Loading…
Cancel
Save