parser/parser/tensorflow/scope/scope_pass_manager.cc

/**
 * Copyright 2020 Huawei Technologies Co., Ltd
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#include "parser/tensorflow/scope/scope_pass_manager.h"
#include "parser/common/acl_graph_parser_util.h"
#include "common/util.h"
#include "common/util/error_manager/error_manager.h"
#include "framework/common/debug/ge_log.h"
#include "register/scope/scope_graph_impl.h"
#include "register/scope/scope_pass_impl.h"

namespace ge {
std::shared_ptr<ScopeGraph> ScopePassManager::BuildScopeGraph(domi::tensorflow::GraphDef *graph_def) {
  GE_CHK_BOOL_EXEC(graph_def != nullptr, return nullptr, "graph_def is nullptr");
  scope_graph_ = ge::parser::MakeShared<ScopeGraph>();
  if (scope_graph_ == nullptr) {
    REPORT_CALL_ERROR("E19999", "New ScopeGraph failed");
    GELOGE(FAILED, "Scope graph make shared failed.");
    return nullptr;
  }
  Status ret = scope_graph_->Init();
  if (ret != SUCCESS) {
    GELOGE(FAILED, "Scope graph init failed.");
    return nullptr;
  }

  auto &impl = scope_graph_->impl_;
  impl->BuildScopeGraph(graph_def);

  return scope_graph_;
}

Status ScopePassManager::AddPass(std::unique_ptr<ScopeBasePass> &pass) {
  GE_CHECK_NOTNULL(pass);

  graph_passes_.push_back(std::move(pass));

  return SUCCESS;
}

Status ScopePassManager::Run(std::shared_ptr<ScopeGraph> &graph) {
  GE_CHECK_NOTNULL(graph);
  bool not_changed = true;

  for (auto &pass : graph_passes_) {
    GE_CHECK_NOTNULL(pass);
    auto &impl = pass->impl_;
    if (impl == nullptr) {
      GELOGE(ge::MEMALLOC_FAILED, "ScopeBasePass is not properly initialized.");
      REPORT_INNER_ERROR("E19999", "ScopeBasePass is not properly initialized");
      continue;
    }

    Status status = impl->Run(graph);
    if (status == SUCCESS) {
      GELOGI("Run scope pass:%s success.", pass->PassName().c_str());
      not_changed = false;
    } else if (status != domi::SCOPE_NOT_CHANGED) {
      // exception
      REPORT_CALL_ERROR("E19999", "Run scope fusion pass [%s] failed.", pass->PassName().c_str());
      GELOGE(FAILED, "Pass Run failed, pass name:%s", pass->PassName().c_str());
      return status;
    }
  }

  return not_changed ? domi::SCOPE_NOT_CHANGED : SUCCESS;
}
}  // namespace ge