Browse Source

Pre Merge pull request !164 from zhengyuanhua/development

pull/164/MERGE
zhengyuanhua Gitee 4 years ago
parent
commit
2bcc35d86e
2 changed files with 2 additions and 1 deletions
  1. +2
    -0
      ge/graph/load/new_model_manager/davinci_model.cc
  2. +0
    -1
      ge/graph/preprocess/insert_op/util_insert_aipp_op.cc

+ 2
- 0
ge/graph/load/new_model_manager/davinci_model.cc View File

@@ -2050,6 +2050,7 @@ Status DavinciModel::SinkModelProfile() {
std::set<uint32_t> task_id_set; std::set<uint32_t> task_id_set;
for (int32_t i = 0; i < task_num; i++) { for (int32_t i = 0; i < task_num; i++) {
auto task = task_list_[i]; auto task = task_list_[i];
GE_CHECK_NOTNULL(task);
auto fusion_op_info = task->GetFusionOpInfo(); auto fusion_op_info = task->GetFusionOpInfo();
// when type is RT_MODEL_TASK_KERNEL, ctx is not null // when type is RT_MODEL_TASK_KERNEL, ctx is not null
if (fusion_op_info != nullptr) { if (fusion_op_info != nullptr) {
@@ -2076,6 +2077,7 @@ Status DavinciModel::SinkModelProfile() {
using Range = std::pair<CIT, CIT>; using Range = std::pair<CIT, CIT>;
for (int32_t i = 0; i < task_num; i++) { for (int32_t i = 0; i < task_num; i++) {
auto task = task_list_[i]; auto task = task_list_[i];
GE_CHECK_NOTNULL(task);
auto fusion_op_info = task->GetFusionOpInfo(); auto fusion_op_info = task->GetFusionOpInfo();
if (fusion_op_info != nullptr && fusion_op_info->original_op_names.size() > 0) { if (fusion_op_info != nullptr && fusion_op_info->original_op_names.size() > 0) {
uint32_t task_id = task->GetTaskID(); uint32_t task_id = task->GetTaskID();


+ 0
- 1
ge/graph/preprocess/insert_op/util_insert_aipp_op.cc View File

@@ -34,7 +34,6 @@
#include "graph/utils/op_desc_utils.h" #include "graph/utils/op_desc_utils.h"
#include "graph/utils/tensor_utils.h" #include "graph/utils/tensor_utils.h"
#include "graph/utils/type_utils.h" #include "graph/utils/type_utils.h"
#include "util_insert_aipp_op.h"


using domi::AippOpParams; using domi::AippOpParams;




Loading…
Cancel
Save