#include "tensorflow/compiler/mlir/tensorflow/transforms/graph_optimization_pass.h"

namespace tensorflow {
namespace {
constexpr int kMlirGraphOptimizationPriority = 0;
}

static mlir_pass_registration::MlirOptimizationPassRegistration
    register_mlir_graph_optimization_pass(
        kMlirGraphOptimizationPriority,
        std::make_unique<mlir::TF::MlirGraphOptimizationPass>());

// corresponding registry.
// -------------------------------------------------------------------------- //

namespace mlir_pass_registration {

class MlirOptimizationPassRegistration {
 public:
  explicit MlirOptimizationPassRegistration(
      int priority, std::unique_ptr<MlirOptimizationPass> pass) {
    MlirOptimizationPassRegistry::Global().Add(priority, std::move(pass));
  }
};

  LOG_FIRST_N(INFO, 1) << "Finish Graph Decomposition Passes";

  return absl::OkStatus();
}

namespace {
constexpr int kMlirGraphDecomposePassPriority = -1;

static mlir_pass_registration::MlirOptimizationPassRegistration
    register_mlir_graph_decompose_pass(kMlirGraphDecomposePassPriority,
                                       std::make_unique<GraphDecomposePass>());
}  // namespace

}  // namespace tfr