2020-03-20 22:40:51 +08:00
|
|
|
set(LLVM_TARGET_DEFINITIONS ONNXOps.td)
|
2020-03-21 00:04:22 +08:00
|
|
|
onnx_mlir_tablegen(ONNXOps.hpp.inc -gen-op-decls "-I${ONNX_MLIR_SRC_ROOT}/compiler/pass")
|
|
|
|
onnx_mlir_tablegen(ONNXOps.cpp.inc -gen-op-defs "-I${ONNX_MLIR_SRC_ROOT}/compiler/pass")
|
2020-03-20 22:40:51 +08:00
|
|
|
set(GEN_DOC_FILE ${CMAKE_BINARY_DIR}/docs/Dialects/onnx.md)
|
|
|
|
add_public_tablegen_target(OMONNXOpsIncGen)
|
|
|
|
|
2020-06-09 10:38:32 +08:00
|
|
|
# Header dependencies target for ONNXOps.hpp
|
|
|
|
add_custom_target(OMONNXOpsInc
|
|
|
|
DEPENDS OMONNXOpsIncGen
|
|
|
|
OMPromotableConstOperandsOpInterfaceIncGen
|
|
|
|
OMResultTypeInferenceOpInterfaceIncGen
|
|
|
|
ShapeInferenceOpInterfaceIncGen)
|
|
|
|
|
2020-03-20 22:40:51 +08:00
|
|
|
add_library(OMONNXOps
|
|
|
|
ONNXOps.cpp
|
2020-05-14 17:31:33 +08:00
|
|
|
ONNXOps.hpp
|
|
|
|
ONNXOpsHelper.cpp
|
|
|
|
ONNXOpsHelper.hpp)
|
2020-03-20 22:40:51 +08:00
|
|
|
target_include_directories(OMONNXOps
|
|
|
|
PRIVATE
|
|
|
|
${ONNX_MLIR_SRC_ROOT}
|
|
|
|
${ONNX_MLIR_BIN_ROOT}
|
|
|
|
${ONNX_MLIR_SRC_ROOT})
|
2020-06-09 10:38:32 +08:00
|
|
|
|
|
|
|
add_dependencies(OMONNXOps OMONNXOpsIncGen)
|
2020-06-04 21:05:04 +08:00
|
|
|
target_link_libraries(OMONNXOps
|
|
|
|
onnx)
|
2020-04-14 17:40:05 +08:00
|
|
|
# Linking dependencies:
|
|
|
|
add_dependencies(OMONNXOps
|
2020-03-20 22:40:51 +08:00
|
|
|
OMPromotableConstOperandsOpInterface
|
2020-05-26 09:54:19 +08:00
|
|
|
OMResultTypeInferenceOpInterface
|
2020-03-20 22:40:51 +08:00
|
|
|
OMShapeInferenceOpInterface)
|
|
|
|
|
2020-03-21 00:04:22 +08:00
|
|
|
add_onnx_mlir_dialect_doc(onnx ONNXOps.td)
|