mindspore/mindspore/ccsrc/optimizer/py_pass_manager.h

/**
 * 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.
 */
#ifndef MINDSPORE_CCSRC_OPTIMIZER_PY_PASS_MANAGER_H_
#define MINDSPORE_CCSRC_OPTIMIZER_PY_PASS_MANAGER_H_

#include <memory>
#include <string>
#include <vector>
#include <unordered_map>

#include "ir/anf.h"
#include "ir/func_graph.h"
#include "ir/primitive.h"
#include "utils/graph_utils.h"
#include "common/utils.h"

#include "pipeline/parse/resolve.h"
#include "optimizer/py_pass.h"
#include "optimizer/pass_group.h"

namespace mindspore {
namespace opt {
namespace python_pass {
class PyPassManager;
using PyPassManagerPtr = std::shared_ptr<PyPassManager>;

enum Phase { RESOLVE, OPT };

class PyPassManager {
 protected:
  PyPassManager();
  static PyPassManagerPtr global_instance;

 public:
  // Singletons should not be cloneable and assignable
  PyPassManager(const PyPassManager &other) = delete;
  void operator=(const PyPassManager &) = delete;
  // Access the only global instance
  static PyPassManagerPtr GetInstance();
  virtual ~PyPassManager() = default;
  void Registe(const std::string &pass_name, const py::function &pattern, const py::function &target,
               Phase phase = Phase::RESOLVE, bool run_only_once = false, bool multigraph = true);
  void Unregiste(const std::string &pass_name, Phase phase);
  PassGroupPtr GetPassGroup(Phase phase);
  void ClearRes();

 private:
  static std::unordered_map<Phase, PassGroupPtr> phase_to_group_;
};
}  // namespace python_pass
}  // namespace opt
}  // namespace mindspore
#endif  // MINDSPORE_CCSRC_OPTIMIZER_PY_PASS_MANAGER_H_