Browse Source

Add ut.

tags/v1.3.0
unknown 4 years ago
parent
commit
fc4b153af0
1 changed files with 2 additions and 2 deletions
  1. +2
    -2
      ge/single_op/single_op_model.cc

+ 2
- 2
ge/single_op/single_op_model.cc View File

@@ -61,7 +61,7 @@ Status IfInferDepend(GeModelPtr &ge_model, bool &flag) {

Status NeedHybridModel(GeModelPtr &ge_model, bool &flag) {
bool infer_depend_flag = false;
GE_CHK_STATUS_RET(IfInferDepend(ge_model, infer_depend_flag), "[Check][IfInferDepend] failed.");
GE_CHK_STATUS_RET(IfInferDepend(ge_model, infer_depend_flag), "[Check][InferDepend] failed.");
auto tasks = ge_model->GetModelTaskDefPtr()->task();
int32_t kernel_task_num = 0;
for (int i = 0; i < tasks.size(); ++i) {
@@ -531,7 +531,7 @@ Status SingleOpModel::BuildDynamicOp(StreamResource &resource, DynamicSingleOp &
auto ge_model = model_helper_.GetGeModel();
GE_CHECK_NOTNULL(ge_model);
bool need_hybrid_model = false;
GE_CHK_STATUS_RET_NOLOG(NeedHybridModel(ge_model, need_hybrid_model), "[Check][NeedHybridModel] failed.");
GE_CHK_STATUS_RET(NeedHybridModel(ge_model, need_hybrid_model), "[Check][NeedHybridModel] failed.");
if (need_hybrid_model) {
GELOGD("Build single op HybridModel.");
GE_CHK_STATUS_RET_NOLOG(hybrid::NodeExecutorManager::GetInstance().EnsureInitialized());


Loading…
Cancel
Save