2020-03-20 22:40:51 +08:00
|
|
|
add_library(OMAttributePromotion
|
|
|
|
AttributePromotion.cpp)
|
|
|
|
target_include_directories(OMAttributePromotion
|
|
|
|
PRIVATE ${ONNX_MLIR_SRC_ROOT} ${ONNX_MLIR_BIN_ROOT}
|
|
|
|
${ONNF_MLIR_SRC_ROOT})
|
2020-04-14 17:40:05 +08:00
|
|
|
|
|
|
|
# Linking dependencies:
|
|
|
|
add_dependencies(OMAttributePromotion
|
2020-03-20 22:40:51 +08:00
|
|
|
OMPromotableConstOperandsOpInterface)
|
|
|
|
|
2020-04-25 04:15:36 +08:00
|
|
|
add_library(OMElideConstants
|
|
|
|
ElideConstants.cpp)
|
|
|
|
target_include_directories(OMElideConstants
|
|
|
|
PRIVATE ${ONNX_MLIR_SRC_ROOT} ${ONNX_MLIR_BIN_ROOT}
|
|
|
|
${ONNF_MLIR_SRC_ROOT})
|
|
|
|
|
2020-03-20 22:40:51 +08:00
|
|
|
set(LLVM_TARGET_DEFINITIONS ONNXRewrite.td)
|
|
|
|
onnx_mlir_tablegen(ONNXRewrite.inc -gen-rewriters)
|
|
|
|
add_public_tablegen_target(OMONNXRewriteIncGen)
|
|
|
|
|
|
|
|
set(LLVM_TARGET_DEFINITIONS ONNXCombine.td)
|
|
|
|
onnx_mlir_tablegen(ONNXCombine.inc -gen-rewriters)
|
|
|
|
add_public_tablegen_target(OMONNXCombineIncGen)
|
|
|
|
|
|
|
|
set(LLVM_TARGET_DEFINITIONS ONNXDecompose.td)
|
|
|
|
onnx_mlir_tablegen(ONNXDecompose.inc -gen-rewriters)
|
|
|
|
add_public_tablegen_target(OMONNXDecomposeIncGen)
|
|
|
|
|
|
|
|
add_library(OMONNXRewrite
|
|
|
|
ONNXRewrite.cpp
|
|
|
|
ONNXCombine.cpp
|
|
|
|
ONNXDecompose.cpp)
|
|
|
|
target_include_directories(OMONNXRewrite
|
|
|
|
PRIVATE ${ONNX_MLIR_SRC_ROOT} ${ONNX_MLIR_BIN_ROOT}
|
|
|
|
${ONNF_MLIR_SRC_ROOT})
|
|
|
|
add_dependencies(OMONNXRewrite
|
|
|
|
OMONNXRewriteIncGen
|
|
|
|
OMONNXDecomposeIncGen
|
|
|
|
OMONNXCombineIncGen)
|
2020-04-14 17:40:05 +08:00
|
|
|
# Linking dependencies:
|
|
|
|
add_dependencies(OMONNXRewrite
|
|
|
|
OMONNXOps)
|
2020-03-20 22:40:51 +08:00
|
|
|
|
|
|
|
add_library(OMShapeInference ShapeInferencePass.cpp)
|
|
|
|
target_include_directories(OMShapeInference
|
|
|
|
PRIVATE ${ONNX_MLIR_SRC_ROOT} ${ONNX_MLIR_BIN_ROOT}
|
|
|
|
${ONNX_MLIR_SRC_ROOT})
|
2020-04-14 17:40:05 +08:00
|
|
|
add_dependencies(OMShapeInference ShapeInferenceOpInterfaceIncGen)
|
|
|
|
# Linking dependencies:
|
|
|
|
add_dependencies(OMShapeInference
|
2020-03-20 22:40:51 +08:00
|
|
|
OMShapeInferenceOpInterface)
|