NiccoloN
0c7db55a24
Validate Operations / validate-operations (push) Has been cancelled
fast pim code emission
54 lines
2.5 KiB
C++
54 lines
2.5 KiB
C++
#include "src/Accelerators/PIM/Compiler/PimCompilerOptions.hpp"
|
|
|
|
#define DEBUG_TYPE "PimCompilerOptions"
|
|
|
|
namespace onnx_mlir {
|
|
|
|
llvm::cl::opt<PimEmissionTargetType> pimEmissionTarget(
|
|
llvm::cl::desc("[Optional] Choose PIM-related target to emit (once selected it will cancel the other targets):"),
|
|
llvm::cl::values(clEnumVal(EmitSpatial, "Lower model to spatial IR")),
|
|
llvm::cl::values(clEnumVal(EmitPim, "Lower model to PIM IR")),
|
|
llvm::cl::values(clEnumVal(EmitPimBufferized, "Lower model to PIM IR and bufferize it")),
|
|
llvm::cl::values(clEnumVal(EmitPimCodegen, "Lower model to PIM IR and generate code for PIM")),
|
|
llvm::cl::init(EmitPimCodegen),
|
|
llvm::cl::cat(OnnxMlirOptions));
|
|
|
|
llvm::cl::opt<bool>
|
|
pimOnlyCodegen("pim-only-codegen",
|
|
llvm::cl::desc("Only generate code for PIM (assume input is already in bufferized PIM IR)"),
|
|
llvm::cl::init(false),
|
|
llvm::cl::cat(OnnxMlirOptions));
|
|
|
|
llvm::cl::opt<bool> useExperimentalConvImpl("use-experimental-conv-impl",
|
|
llvm::cl::desc("Use experimental implementation for convolution"),
|
|
llvm::cl::init(false),
|
|
llvm::cl::cat(OnnxMlirOptions));
|
|
|
|
llvm::cl::opt<bool> pimEmitJson("pim-emit-json",
|
|
llvm::cl::desc("Also emit per-core JSON instruction files alongside binary .pim files"),
|
|
llvm::cl::init(false),
|
|
llvm::cl::cat(OnnxMlirOptions));
|
|
|
|
llvm::cl::opt<size_t>
|
|
crossbarSize("crossbar-size", llvm::cl::desc("Width and heigth of a single crossbar"), llvm::cl::init(2));
|
|
|
|
llvm::cl::opt<size_t>
|
|
crossbarCountInCore("crossbar-count", llvm::cl::desc("Number of crossbars in each core"), llvm::cl::init(256));
|
|
|
|
llvm::cl::opt<long> coresCount("core-count",
|
|
llvm::cl::desc("Number of cores in the chip. `-1` to use the minimum amount of cores."),
|
|
llvm::cl::init(-1));
|
|
|
|
llvm::cl::opt<size_t> dcpCriticalWindowSize(
|
|
"dcp-critical-window-size",
|
|
llvm::cl::desc("Number of lowest-slack virtual nodes considered by each DCP coarsening iteration. "
|
|
"Use 0 to run the legacy full-graph DCP analysis."),
|
|
llvm::cl::init(4000));
|
|
|
|
llvm::cl::opt<bool>
|
|
ignoreConcatError("ignore-concat-error",
|
|
llvm::cl::desc("Ignore ConcatOp corner case: do not assert and do a simplification"),
|
|
llvm::cl::init(false));
|
|
|
|
} // namespace onnx_mlir
|