54 lines
1.4 KiB
Python
54 lines
1.4 KiB
Python
package(
|
|
default_visibility = ["//visibility:public"],
|
|
)
|
|
|
|
genrule(
|
|
name = "ExternalUtil",
|
|
srcs = [
|
|
"ExternalUtil.hpp.in",
|
|
],
|
|
outs = [
|
|
"ExternalUtil.hpp",
|
|
],
|
|
cmd = "cp $(location :ExternalUtil.hpp.in) $(location :ExternalUtil.hpp)",
|
|
)
|
|
|
|
cc_library(
|
|
name = "MainUtils",
|
|
srcs = ["MainUtils.cpp"],
|
|
hdrs = [
|
|
"MainUtils.hpp",
|
|
":ExternalUtil",
|
|
],
|
|
deps = [
|
|
"//src/Builder:OMBuilder",
|
|
"//src/Dialect/Krnl:OMKrnlOps",
|
|
"//src/Dialect/ONNX:OMONNXOps",
|
|
"//src/Pass:Passes",
|
|
"@llvm-project//mlir:AllPassesAndDialectsNoRegistration",
|
|
"@onnx",
|
|
],
|
|
)
|
|
|
|
cc_binary(
|
|
name = "onnx-mlir",
|
|
srcs = ["main.cpp"],
|
|
copts = [],
|
|
deps = [
|
|
":MainUtils",
|
|
"//src/Conversion/KrnlToAffine:OMKrnlToAffine",
|
|
"//src/Conversion/KrnlToLLVM:OMKrnlToLLVM",
|
|
"//src/Conversion/ONNXToKrnl:OMONNXToKrnl",
|
|
"//src/Transform:OMBundleMemoryPools",
|
|
"//src/Transform:OMDisconnectKrnlDimFromAlloc",
|
|
"//src/Transform:OMElideKrnlGlobalConstants",
|
|
"//src/Transform:OMEnableMemoryPool",
|
|
"//src/Transform:OMLowerKrnlShape",
|
|
"//src/Transform:OMPackKrnlGlobalConstants",
|
|
"//src/Transform/ONNX:OMAttributePromotion",
|
|
"//src/Transform/ONNX:OMElideConstants",
|
|
"//src/Transform/ONNX:OMONNXRewrite",
|
|
"//src/Transform/ONNX:OMShapeInference",
|
|
],
|
|
)
|