// Copyright (c) 2023 PaddlePaddle Authors. All Rights Reserved.
//
// 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.

#pragma once

#include "paddle/pass/pass.h"

namespace ir {

class Operation;

class PassManager;

namespace detail {
// Used to run operation passes over nested operations.
class PassAdaptor final : public Pass {
 public:
  explicit PassAdaptor(PassManager* pm) : Pass("pass_adaptor", 0), pm_(pm) {}

  void Run(ir::Operation*) override {}

  void Run(ir::Operation*, uint8_t opt_level);

 private:
  void RunImpl(ir::Operation* op, uint8_t opt_level);

  static bool RunPass(Pass* pass, ir::Operation* op, uint8_t opt_level);

  static bool RunPipeline(const PassManager& pm,
                          ir::Operation* op,
                          uint8_t opt_level);

  // Use for RunImpl later.
  PassManager* pm_;

  // For accessing RunPipeline.
  friend class ir::PassManager;
};
}  // namespace detail

}  // namespace ir