From a9b0c286713acf50b04beb047c2ec280c978bfed Mon Sep 17 00:00:00 2001 From: unknown Date: Thu, 25 Mar 2021 19:02:45 +0800 Subject: [PATCH] Fix error. --- ge/graph/load/model_manager/davinci_model.cc | 2 +- ge/hybrid/executor/hybrid_model_executor.cc | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/ge/graph/load/model_manager/davinci_model.cc b/ge/graph/load/model_manager/davinci_model.cc index f15c9f49..d7b8e443 100755 --- a/ge/graph/load/model_manager/davinci_model.cc +++ b/ge/graph/load/model_manager/davinci_model.cc @@ -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(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); diff --git a/ge/hybrid/executor/hybrid_model_executor.cc b/ge/hybrid/executor/hybrid_model_executor.cc index 2d729e17..3e65b5c1 100755 --- a/ge/hybrid/executor/hybrid_model_executor.cc +++ b/ge/hybrid/executor/hybrid_model_executor.cc @@ -87,7 +87,7 @@ Status HybridModelExecutor::ProfileStepInfo(uint16_t tag_id) { uint64_t index_id = context_.iteration + 1; uint64_t model_id = static_cast(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_);