#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> createSpatialToGraphvizPass();

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

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

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

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

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

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

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

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

} // namespace onnx_mlir