#pragma once

#include "mlir/Dialect/Func/IR/FuncOps.h"
#include "mlir/IR/MLIRContext.h"

namespace onnx_mlir {

void populateEarlyPostPatterns(mlir::RewritePatternSet& patterns, mlir::MLIRContext* ctx);

void populatePostPatterns(mlir::RewritePatternSet& patterns, mlir::MLIRContext* ctx);

void annotateWeightsConstants(mlir::func::FuncOp funcOp);

} // namespace onnx_mlir