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