#pragma once

#include "mlir/IR/DialectRegistry.h"

#include "src/Accelerators/PIM/Dialect/Pim/PimOps.hpp"

namespace onnx_mlir {
namespace pim {

void registerOpBufferizationInterfaces(mlir::DialectRegistry& registry);

} // namespace pim
} // namespace onnx_mlir
