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)
|
|
|
|
|
|
|
|
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})
|
|
|
|
add_dependencies(OMONNXOps OMONNXOpsIncGen)
|
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)
|