30 lines
1.3 KiB
CMake
30 lines
1.3 KiB
CMake
find_package(Python 3 REQUIRED COMPONENTS Interpreter)
|
|
|
|
# Invoke gen_onnx_mlir.py to obtain ONNXOps.td.inc, OpBuildTable.inc.
|
|
add_custom_command(OUTPUT ${CMAKE_CURRENT_SOURCE_DIR}/ONNXOps.td.inc
|
|
${CMAKE_CURRENT_SOURCE_DIR}/OpBuildTable.inc
|
|
COMMAND Python::Interpreter ${CMAKE_CURRENT_SOURCE_DIR}/gen_onnx_mlir.py
|
|
DEPENDS ${CMAKE_CURRENT_SOURCE_DIR}/gen_onnx_mlir.py)
|
|
|
|
# Move the generated files to respective destinations:
|
|
# ONNXOps.td.inc -> src/Dialect/ONNX/ONNXOps.td.inc
|
|
add_custom_target(OMONNXOpsTableGenIncGen
|
|
COMMAND ${CMAKE_COMMAND} -E rename
|
|
${CMAKE_CURRENT_SOURCE_DIR}/ONNXOps.td.inc
|
|
${ONNX_MLIR_SRC_ROOT}/src/Dialect/ONNX/ONNXOps.td.inc
|
|
DEPENDS ${CMAKE_CURRENT_SOURCE_DIR}/ONNXOps.td.inc)
|
|
|
|
# OpBuildTable.inc -> src/Builder/OpBuildTable.inc
|
|
add_custom_target(OMONNXOpsBuildTableIncGen
|
|
COMMAND ${CMAKE_COMMAND} -E rename
|
|
${CMAKE_CURRENT_SOURCE_DIR}/OpBuildTable.inc
|
|
${ONNX_MLIR_SRC_ROOT}/src/Builder/OpBuildTable.inc
|
|
DEPENDS ${CMAKE_CURRENT_SOURCE_DIR}/OpBuildTable.inc)
|
|
|
|
add_custom_target(OMONNXOpsIncTranslation
|
|
DEPENDS OMONNXOpsTableGenIncGen
|
|
OMONNXOpsBuildTableIncGen)
|
|
|
|
add_custom_target(OMONNXCheckVersion
|
|
COMMAND python ${CMAKE_CURRENT_SOURCE_DIR}/gen_onnx_mlir.py --check-operation-version)
|