NiccoloN
909c4acfdd
Validate Operations / validate-operations (push) Has been cancelled
remove Spatial many ops in favor of tensor ops like in pim
27 lines
1012 B
C++
27 lines
1012 B
C++
#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
|