#pragma once

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

#include "llvm/ADT/StringRef.h"
#include "llvm/Support/JSON.h"

#include "onnx-mlir/Compiler/OMCompilerTypes.h"

namespace onnx_mlir {

class PimAcceleratorMemory;

OnnxMlirCompilerErrorCodes writeHostCoreJson(llvm::StringRef outputDirPath);
OnnxMlirCompilerErrorCodes writeMemoryBinary(mlir::ModuleOp moduleOp,
                                             mlir::func::FuncOp funcOp,
                                             PimAcceleratorMemory& memory,
                                             llvm::StringRef outputDirPath);
OnnxMlirCompilerErrorCodes writeConfigJson(mlir::func::FuncOp funcOp,
                                           PimAcceleratorMemory& memory,
                                           size_t maxCoreId,
                                           llvm::json::Object xbarsPerArrayGroup,
                                           llvm::StringRef outputDirPath);

} // namespace onnx_mlir