llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
torch-mlir/external/iree-dialects/python/CMakeLists.txt

66 lines
2.2 KiB
CMake
Raw Normal View History

Initial import of iree-dialects. We plan on using these dialects "natively" as part of the npcomp backend contract, and provide feedback to evolve them in IREE. Roughly speaking, we can consider these dialects as "what's missing from upstream that we think belongs in the general abstraction layer that npcomp's backend contract targets". We integrate them by just copying the relevant directory from the IREE source tree (with `build_tools/update_iree_dialects.sh`). This avoids adding IREE as a submodule, which is way too heavyweight (including IREE itself, another copy of LLVM, TensorFlow, ...) and would give the false impression of a source dependency rather than the lightweight (and eventually versioned/stabilized) IR-level compatibility that we strive for.
2021-08-12 02:56:25 +08:00
include(AddMLIRPython)
################################################################################
# Sources
################################################################################
declare_mlir_python_sources(IREEDialectsPythonSources)
declare_mlir_python_sources(IREEDialectsPythonExtensions)
declare_mlir_python_sources(IREEDialectsPythonSources.Dialects
ADD_TO_PARENT IREEDialectsPythonSources
)
declare_mlir_dialect_python_bindings(
ADD_TO_PARENT IREEDialectsPythonSources.Dialects
ROOT_DIR "${CMAKE_CURRENT_SOURCE_DIR}/mlir"
TD_FILE dialects/IreeBinding.td
SOURCES dialects/iree.py
DIALECT_NAME iree
)
################################################################################
# Extensions
################################################################################
declare_mlir_python_extension(IREEDialectsPythonExtensions.Main
MODULE_NAME _ireeDialects
ADD_TO_PARENT IREEDialectsPythonExtensions
SOURCES
${CMAKE_CURRENT_SOURCE_DIR}/IREEDialectsModule.cpp
EMBED_CAPI_LINK_LIBS
IREEDialectsCAPI
PRIVATE_LINK_LIBS
LLVMSupport
)
################################################################################
# Generate packages and shared library
# Downstreams typically will not use these, but they are useful for local
# testing.
################################################################################
set(_source_components
# TODO: Core is now implicitly building/registering all dialects, increasing
# build burden by ~5x. Make it stop.
MLIRPythonSources.Core
IREEDialectsPythonSources
IREEDialectsPythonExtensions
)
add_mlir_python_common_capi_library(IREEDialectsAggregateCAPI
INSTALL_COMPONENT IREEDialectsPythonModules
INSTALL_DESTINATION python_packages/iree_dialects/mlir/_mlir_libs
OUTPUT_DIRECTORY "${IREE_DIALECTS_BINARY_DIR}/python_packages/iree_dialects/mlir/_mlir_libs"
RELATIVE_INSTALL_ROOT "../../../.."
DECLARED_SOURCES ${_source_components}
)
add_mlir_python_modules(IREEDialectsPythonModules
ROOT_PREFIX "${IREE_DIALECTS_BINARY_DIR}/python_packages/iree_dialects/mlir"
INSTALL_PREFIX "python_packages/iree_dialects/mlir"
DECLARED_SOURCES ${_source_components}
COMMON_CAPI_LINK_LIBS
IREEDialectsAggregateCAPI
)