Fix error.

4 years ago · a9b0c28671
parent 0da536a15b
commit a9b0c28671
2 changed files with 2 additions and 2 deletions
--- 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<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);
--- 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<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_);