Browse Source

dts: pass interface impl modify

tags/v1.1.0
zhengyuanhua 3 years ago
parent
commit
d83b559024
4 changed files with 4 additions and 3 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
  3. +1
    -1
      metadef
  4. +1
    -1
      parser

+ 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;
for (int32_t i = 0; i < task_num; i++) {
auto task = task_list_[i];
GE_CHECK_NOTNULL(task);
auto fusion_op_info = task->GetFusionOpInfo();
// when type is RT_MODEL_TASK_KERNEL, ctx is not null
if (fusion_op_info != nullptr) {
@@ -2076,6 +2077,7 @@ Status DavinciModel::SinkModelProfile() {
using Range = std::pair<CIT, CIT>;
for (int32_t i = 0; i < task_num; i++) {
auto task = task_list_[i];
GE_CHECK_NOTNULL(task);
auto fusion_op_info = task->GetFusionOpInfo();
if (fusion_op_info != nullptr && fusion_op_info->original_op_names.size() > 0) {
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/tensor_utils.h"
#include "graph/utils/type_utils.h"
#include "util_insert_aipp_op.h"

using domi::AippOpParams;



+ 1
- 1
metadef

@@ -1 +1 @@
Subproject commit cc9de48a7779cf95cab90a23db608421a691fd12
Subproject commit 384dab951f40b90788cc755e13754ed386ccb9c6

+ 1
- 1
parser

@@ -1 +1 @@
Subproject commit f86c751307835d85329260feb51b559d378b5710
Subproject commit f9e30cdc643926d91f7b650b6903a7b3bf6f597b

Loading…
Cancel
Save