| # Generates runtime library resources. | |
| file(GLOB_RECURSE _python_sources | |
| ${IREE_DIALECTS_SOURCE_DIR}/python/iree/compiler/dialects/iree_pydm/*.py | |
| ) | |
| add_custom_target(IREEPyDMRTLGen | |
| DEPENDS | |
| IREEDialectsPythonModules | |
| SOURCES ${_python_sources} | |
| ) | |
| function(rtl_builder) | |
| cmake_parse_arguments(ARG | |
| "" | |
| "MLIR_FILE" | |
| "RTL_MODULES" | |
| ${ARGN}) | |
| add_custom_command( | |
| TARGET IREEPyDMRTLGen PRE_BUILD | |
| BYPRODUCTS "${ARG_MLIR_FILE}" | |
| COMMAND | |
| "${CMAKE_COMMAND}" -E env | |
| "PYTHONPATH=${IREE_DIALECTS_BINARY_DIR}/python_packages/iree_dialects" | |
| "${Python3_EXECUTABLE}" -m iree.compiler.dialects.iree_pydm.rtl.rtl_builder | |
| --output "${ARG_MLIR_FILE}" | |
| ${ARG_RTL_MODULES} | |
| COMMENT "Generating PyDM RTL: ${ARG_MLIR_FILE}" | |
| WORKING_DIRECTORY "${CMAKE_CURRENT_BINARY_DIR}" | |
| ) | |
| endfunction() | |
| rtl_builder( | |
| MLIR_FILE PyDMRTLBase.mlir | |
| RTL_MODULES | |
| .booleans | |
| .numerics | |
| ) |