35 lines
757 B
Python
35 lines
757 B
Python
|
|
||
|
|
cc_binary(
|
||
|
|
name = "MainUtils",
|
||
|
|
srcs = ["MainUtils.hpp","MainUtils.cpp"],
|
||
|
|
copts = [],
|
||
|
|
deps = [
|
||
|
|
"@onnx",
|
||
|
|
"//Builder:OMBuilder",
|
||
|
|
OMKrnlOps
|
||
|
|
OMONNXOps
|
||
|
|
OMKrnlToAffine
|
||
|
|
OMKrnlToLLVM
|
||
|
|
OMONNXToKrnl
|
||
|
|
OMONNXRewrite
|
||
|
|
OMShapeInference
|
||
|
|
OMShapeInferenceOpInterface
|
||
|
|
OMAttributePromotion
|
||
|
|
OMPromotableConstOperandsOpInterface
|
||
|
|
OMResultTypeInferenceOpInterface
|
||
|
|
OMElideConstants
|
||
|
|
OMElideKrnlGlobalConstants
|
||
|
|
OMPackKrnlGlobalConstants
|
||
|
|
OMEnableMemoryPool
|
||
|
|
OMBundleMemoryPools
|
||
|
|
OMDisconnectKrnlDimFromAlloc
|
||
|
|
OMLowerKrnlShape
|
||
|
|
],
|
||
|
|
)
|
||
|
|
|
||
|
|
cc_binary(
|
||
|
|
name = "onnx-mlir",
|
||
|
|
srcs = ["main.cpp"],
|
||
|
|
copts = [],
|
||
|
|
deps = [":MainUtils"],
|
||
|
|
)
|