colin
/
onnx-mlir
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity
onnx-mlir/src/Dialect/ONNX/CMakeLists.txt

20 lines
728 B
CMake
Raw Normal View History

[NFC] Breakdown CMake Build Scripts to Smaller Parts (#39) * 1.Break down CMake scripts to smaller libraries per-directory. 2. Move some transformations and interfaces to the right folder. 3. Fix minor merge failure of the patch renaming files to use LLVM convention. * Link OMBuilder with OMONNXOps. * 1. Update the src location of generated ONNX dialect definition. 2. Link OMONNXRewrite with OMONNXOps. * Fix path to tablegen for add_onnx_mlir_dialect_doc. * Update build script for onnx_mlir_transform. * 1. Remove comment code. 2. onnx_mlir_attribute_promotion -> OMAttributePromotion. * Name tablegen generated files with LLVM convention. * Nit: reorder libraries to link against. * Nit: Link against MLIR first.
2020-03-20 22:40:51 +08:00
set(LLVM_TARGET_DEFINITIONS ONNXOps.td)
onnx_mlir_tablegen(ONNXOps.hpp.inc -gen-op-decls "-I${CMAKE_SOURCE_DIR}/compiler/pass")
onnx_mlir_tablegen(ONNXOps.cpp.inc -gen-op-defs "-I${CMAKE_SOURCE_DIR}/compiler/pass")
set(GEN_DOC_FILE ${CMAKE_BINARY_DIR}/docs/Dialects/onnx.md)
add_public_tablegen_target(OMONNXOpsIncGen)
add_library(OMONNXOps
ONNXOps.cpp
ONNXOps.hpp)
target_include_directories(OMONNXOps
PRIVATE
${ONNX_MLIR_SRC_ROOT}
${ONNX_MLIR_BIN_ROOT}
${ONNX_MLIR_SRC_ROOT})
add_dependencies(OMONNXOps OMONNXOpsIncGen)
target_link_libraries(OMONNXOps
OMPromotableConstOperandsOpInterface
OMShapeInferenceOpInterface)
add_onnx_mlir_dialect_doc(onnx ONNXOps.td)