#pragma once

#include "mlir/Pass/Pass.h"

#include <memory>
#include <string>

namespace onnx_mlir {

std::unique_ptr<mlir::Pass> createONNXToSpatialPass();
std::unique_ptr<mlir::Pass> createSpatialLayoutPlanningPass();
std::unique_ptr<mlir::Pass> createLowerSpatialPlansPass();

std::unique_ptr<mlir::Pass> createSpatialToPimPass();

std::unique_ptr<mlir::Pass> createPimBufferizationPass();

std::unique_ptr<mlir::Pass> createPimMemoryCoalescingPass();

std::unique_ptr<mlir::Pass> createMergeComputeNodesPass();

std::unique_ptr<mlir::Pass> createPimHostConstantFoldingPass();

std::unique_ptr<mlir::Pass> createPimVerificationPass();

std::unique_ptr<mlir::Pass> createEmitPimCodePass();

std::unique_ptr<mlir::Pass> createMessagePass(std::string message);

} // namespace onnx_mlir