Browse Source

Fix error.

pull/1329/head
unknown 4 years ago
parent
commit
a9b0c28671
2 changed files with 2 additions and 2 deletions
  1. +1
    -1
      ge/graph/load/model_manager/davinci_model.cc
  2. +1
    -1
      ge/hybrid/executor/hybrid_model_executor.cc

+ 1
- 1
ge/graph/load/model_manager/davinci_model.cc View File

@@ -2313,7 +2313,7 @@ Status DavinciModel::ProfileStepInfo(DavinciModel *model, uint16_t tag_id) {
uint64_t index_id = model->iterator_count_ + 1;
uint64_t model_id = static_cast<uint64_t>(model->Id());
rtError_t rt_ret = RT_ERROR_NONE;
#ifdef ENABLE_OPEN_SRC
#ifndef ONLY_COMPILE_OPEN_SRC
rtStream_t stream = model->rt_model_stream_;
GELOGD("Profiling Step Info TraceTask execute async start, index_id = %lu, model_id = %lu, tag_id = %u",
index_id, model_id, tag_id);


+ 1
- 1
ge/hybrid/executor/hybrid_model_executor.cc View File

@@ -87,7 +87,7 @@ Status HybridModelExecutor::ProfileStepInfo(uint16_t tag_id) {
uint64_t index_id = context_.iteration + 1;
uint64_t model_id = static_cast<uint64_t>(model_->GetModelId());
rtError_t rt_ret = RT_ERROR_NONE;
#ifdef ENABLE_OPEN_SRC
#ifndef ONLY_COMPILE_OPEN_SRC
GELOGD("Profiling Step Info TraceTask execute async start, index_id = %lu, model_id = %lu, tag_id = %u",
index_id, model_id, tag_id);
rt_ret = rtProfilerTraceEx(index_id, model_id, tag_id, stream_);


Loading…
Cancel
Save