Browse Source

update new line

pull/616/head
taoxiangdong 5 years ago
parent
commit
66ab5d7dae
2 changed files with 6 additions and 3 deletions
  1. +4
    -2
      ge/graph/build/model_builder.h
  2. +2
    -1
      ge/graph/load/new_model_manager/model_manager.h

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

@@ -83,9 +83,11 @@ class ModelBuilder {

Status CompileSingleOp();

Status CollectCheckAicpuAttr(const OpDescPtr &op_desc, std::set<std::string> &cpue_check_set, std::set<std::string> &tf_engine_set);
Status CollectCheckAicpuAttr(const OpDescPtr &op_desc, std::set<std::string> &cpue_check_set,
std::set<std::string> &tf_engine_set);

Status SetModelCheckAicpuAttr(ge::Model &model, std::set<std::string> &aicpu_optype_set, std::set<std::string> &aicpu_tf_optype_set);
Status SetModelCheckAicpuAttr(ge::Model &model, std::set<std::string> &aicpu_optype_set,
std::set<std::string> &aicpu_tf_optype_set);

uint64_t session_id_;



+ 2
- 1
ge/graph/load/new_model_manager/model_manager.h View File

@@ -295,7 +295,8 @@ class FMK_FUNC_HOST_VISIBILITY FMK_FUNC_DEV_VISIBILITY ModelManager {

ge::Status LaunchKernelCustAicpuSo(const string &kernel_name);

ge::Status LaunchKernelCheckAicpuOp(std::vector<std::string> &aicpu_optype_list, std::vector<std::string> &aicpu_tf_optype_list);
ge::Status LaunchKernelCheckAicpuOp(std::vector<std::string> &aicpu_optype_list,
std::vector<std::string> &aicpu_tf_optype_list);

ge::Status CheckAicpuOpList(GeModelPtr ge_model);



Loading…
Cancel
Save