| include(AddMLIRPython) |
| |
| ################################################################################ |
| # Structural groupings. |
| ################################################################################ |
| |
| declare_mlir_python_sources(MLIRPythonSources) |
| declare_mlir_python_sources(MLIRPythonSources.Dialects |
| ADD_TO_PARENT MLIRPythonSources) |
| declare_mlir_python_sources(MLIRPythonSources.Core |
| ADD_TO_PARENT MLIRPythonSources) |
| |
| ################################################################################ |
| # Pure python sources and generated code |
| ################################################################################ |
| |
| declare_mlir_python_sources(MLIRPythonSources.Core.Python |
| ROOT_DIR "${CMAKE_CURRENT_SOURCE_DIR}/mlir" |
| ADD_TO_PARENT MLIRPythonSources.Core |
| SOURCES |
| _mlir_libs/__init__.py |
| ir.py |
| passmanager.py |
| rewrite.py |
| dialects/_ods_common.py |
| |
| # The main _mlir module has submodules: include stubs from each. |
| _mlir_libs/_mlir/__init__.pyi |
| _mlir_libs/_mlir/ir.pyi |
| _mlir_libs/_mlir/passmanager.pyi |
| ) |
| |
| declare_mlir_python_sources(MLIRPythonSources.Core.Python.Extras |
| ROOT_DIR "${CMAKE_CURRENT_SOURCE_DIR}/mlir" |
| ADD_TO_PARENT MLIRPythonSources.Core.Python |
| SOURCES |
| extras/types.py |
| extras/meta.py |
| ) |
| |
| declare_mlir_python_sources(MLIRPythonSources.ExecutionEngine |
| ROOT_DIR "${CMAKE_CURRENT_SOURCE_DIR}/mlir" |
| ADD_TO_PARENT MLIRPythonSources |
| SOURCES |
| execution_engine.py |
| _mlir_libs/_mlirExecutionEngine.pyi |
| SOURCES_GLOB |
| runtime/*.py |
| ) |
| |
| declare_mlir_python_sources(MLIRPythonSources.Utils |
| ROOT_DIR "${CMAKE_CURRENT_SOURCE_DIR}/mlir" |
| ADD_TO_PARENT MLIRPythonSources |
| SOURCES |
| utils.py |
| ) |
| |
| declare_mlir_python_sources(MLIRPythonCAPI.HeaderSources |
| ROOT_DIR "${MLIR_SOURCE_DIR}/include" |
| SOURCES_GLOB "mlir-c/*.h" |
| ) |
| |
| ################################################################################ |
| # Dialect bindings |
| ################################################################################ |
| |
| declare_mlir_dialect_python_bindings( |
| ADD_TO_PARENT MLIRPythonSources.Dialects |
| ROOT_DIR "${CMAKE_CURRENT_SOURCE_DIR}/mlir" |
| TD_FILE dialects/AffineOps.td |
| SOURCES |
| dialects/affine.py |
| DIALECT_NAME affine |
| GEN_ENUM_BINDINGS) |
| |
| declare_mlir_dialect_python_bindings( |
| ADD_TO_PARENT MLIRPythonSources.Dialects |
| ROOT_DIR "${CMAKE_CURRENT_SOURCE_DIR}/mlir" |
| TD_FILE dialects/AMDGPUOps.td |
| SOURCES |
| dialects/amdgpu.py |
| DIALECT_NAME amdgpu |
| GEN_ENUM_BINDINGS) |
| |
| declare_mlir_dialect_python_bindings( |
| ADD_TO_PARENT MLIRPythonSources.Dialects |
| ROOT_DIR "${CMAKE_CURRENT_SOURCE_DIR}/mlir" |
| TD_FILE dialects/AsyncOps.td |
| SOURCES_GLOB dialects/async_dialect/*.py |
| DIALECT_NAME async) |
| |
| declare_mlir_dialect_python_bindings( |
| ADD_TO_PARENT MLIRPythonSources.Dialects |
| ROOT_DIR "${CMAKE_CURRENT_SOURCE_DIR}/mlir" |
| TD_FILE dialects/BufferizationOps.td |
| SOURCES |
| dialects/bufferization.py |
| DIALECT_NAME bufferization |
| GEN_ENUM_BINDINGS_TD_FILE |
| "../../include/mlir/Dialect/Bufferization/IR/BufferizationEnums.td" |
| ) |
| |
| declare_mlir_dialect_python_bindings( |
| ADD_TO_PARENT MLIRPythonSources.Dialects |
| ROOT_DIR "${CMAKE_CURRENT_SOURCE_DIR}/mlir" |
| TD_FILE dialects/BuiltinOps.td |
| SOURCES |
| dialects/builtin.py |
| DIALECT_NAME builtin) |
| |
| declare_mlir_dialect_python_bindings( |
| ADD_TO_PARENT MLIRPythonSources.Dialects |
| ROOT_DIR "${CMAKE_CURRENT_SOURCE_DIR}/mlir" |
| TD_FILE dialects/ComplexOps.td |
| SOURCES |
| dialects/complex.py |
| DIALECT_NAME complex) |
| |
| declare_mlir_dialect_python_bindings( |
| ADD_TO_PARENT MLIRPythonSources.Dialects |
| ROOT_DIR "${CMAKE_CURRENT_SOURCE_DIR}/mlir" |
| TD_FILE dialects/IndexOps.td |
| SOURCES |
| dialects/index.py |
| DIALECT_NAME index |
| GEN_ENUM_BINDINGS) |
| |
| declare_mlir_dialect_python_bindings( |
| ADD_TO_PARENT MLIRPythonSources.Dialects |
| ROOT_DIR "${CMAKE_CURRENT_SOURCE_DIR}/mlir" |
| TD_FILE dialects/ControlFlowOps.td |
| SOURCES |
| dialects/cf.py |
| DIALECT_NAME cf) |
| |
| declare_mlir_dialect_python_bindings( |
| ADD_TO_PARENT MLIRPythonSources.Dialects |
| ROOT_DIR "${CMAKE_CURRENT_SOURCE_DIR}/mlir" |
| TD_FILE dialects/FuncOps.td |
| SOURCES |
| dialects/func.py |
| DIALECT_NAME func) |
| |
| declare_mlir_dialect_python_bindings( |
| ADD_TO_PARENT MLIRPythonSources.Dialects |
| ROOT_DIR "${CMAKE_CURRENT_SOURCE_DIR}/mlir" |
| TD_FILE dialects/GPUOps.td |
| SOURCES_GLOB dialects/gpu/*.py |
| DIALECT_NAME gpu |
| GEN_ENUM_BINDINGS) |
| |
| declare_mlir_dialect_python_bindings( |
| ADD_TO_PARENT MLIRPythonSources.Dialects |
| ROOT_DIR "${CMAKE_CURRENT_SOURCE_DIR}/mlir" |
| TD_FILE dialects/LinalgOps.td |
| SOURCES |
| SOURCES_GLOB |
| dialects/linalg/*.py |
| DIALECT_NAME linalg |
| DEPENDS LinalgOdsGen |
| GEN_ENUM_BINDINGS) |
| |
| declare_mlir_dialect_python_bindings( |
| ADD_TO_PARENT MLIRPythonSources.Dialects |
| ROOT_DIR "${CMAKE_CURRENT_SOURCE_DIR}/mlir" |
| TD_FILE dialects/LLVMOps.td |
| SOURCES |
| dialects/llvm.py |
| DIALECT_NAME llvm |
| GEN_ENUM_BINDINGS) |
| |
| declare_mlir_dialect_extension_python_bindings( |
| ADD_TO_PARENT MLIRPythonSources.Dialects |
| ROOT_DIR "${CMAKE_CURRENT_SOURCE_DIR}/mlir" |
| TD_FILE dialects/TransformPDLExtensionOps.td |
| SOURCES |
| dialects/transform/pdl.py |
| DIALECT_NAME transform |
| EXTENSION_NAME transform_pdl_extension) |
| |
| declare_mlir_dialect_extension_python_bindings( |
| ADD_TO_PARENT MLIRPythonSources.Dialects |
| ROOT_DIR "${CMAKE_CURRENT_SOURCE_DIR}/mlir" |
| TD_FILE dialects/TransformDebugExtensionOps.td |
| SOURCES |
| dialects/transform/debug.py |
| DIALECT_NAME transform |
| EXTENSION_NAME transform_debug_extension) |
| |
| declare_mlir_dialect_python_bindings( |
| ADD_TO_PARENT MLIRPythonSources.Dialects |
| ROOT_DIR "${CMAKE_CURRENT_SOURCE_DIR}/mlir" |
| TD_FILE dialects/TransformOps.td |
| SOURCES |
| dialects/transform/__init__.py |
| _mlir_libs/_mlir/dialects/transform/__init__.pyi |
| DIALECT_NAME transform |
| GEN_ENUM_BINDINGS_TD_FILE |
| "../../include/mlir/Dialect/Transform/IR/TransformAttrs.td" |
| ) |
| |
| declare_mlir_python_sources( |
| MLIRPythonSources.Dialects.transform.extras |
| ADD_TO_PARENT MLIRPythonSources.Dialects |
| ROOT_DIR "${CMAKE_CURRENT_SOURCE_DIR}/mlir" |
| GEN_ENUM_BINDINGS |
| SOURCES |
| dialects/transform/extras/__init__.py) |
| |
| declare_mlir_python_sources( |
| MLIRPythonSources.Dialects.transform.interpreter |
| ADD_TO_PARENT MLIRPythonSources.Dialects |
| ROOT_DIR "${CMAKE_CURRENT_SOURCE_DIR}/mlir" |
| SOURCES |
| dialects/transform/interpreter/__init__.py) |
| |
| declare_mlir_dialect_extension_python_bindings( |
| ADD_TO_PARENT MLIRPythonSources.Dialects |
| ROOT_DIR "${CMAKE_CURRENT_SOURCE_DIR}/mlir" |
| TD_FILE dialects/BufferizationTransformOps.td |
| SOURCES |
| dialects/transform/bufferization.py |
| DIALECT_NAME transform |
| EXTENSION_NAME bufferization_transform) |
| |
| declare_mlir_dialect_extension_python_bindings( |
| ADD_TO_PARENT MLIRPythonSources.Dialects |
| ROOT_DIR "${CMAKE_CURRENT_SOURCE_DIR}/mlir" |
| TD_FILE dialects/GPUTransformOps.td |
| SOURCES |
| dialects/transform/gpu.py |
| DIALECT_NAME transform |
| EXTENSION_NAME gpu_transform) |
| |
| declare_mlir_dialect_extension_python_bindings( |
| ADD_TO_PARENT MLIRPythonSources.Dialects |
| ROOT_DIR "${CMAKE_CURRENT_SOURCE_DIR}/mlir" |
| TD_FILE dialects/SCFLoopTransformOps.td |
| SOURCES |
| dialects/transform/loop.py |
| DIALECT_NAME transform |
| EXTENSION_NAME loop_transform) |
| |
| declare_mlir_dialect_extension_python_bindings( |
| ADD_TO_PARENT MLIRPythonSources.Dialects |
| ROOT_DIR "${CMAKE_CURRENT_SOURCE_DIR}/mlir" |
| TD_FILE dialects/MemRefTransformOps.td |
| SOURCES |
| dialects/transform/memref.py |
| DIALECT_NAME transform |
| EXTENSION_NAME memref_transform) |
| |
| declare_mlir_dialect_extension_python_bindings( |
| ADD_TO_PARENT MLIRPythonSources.Dialects |
| ROOT_DIR "${CMAKE_CURRENT_SOURCE_DIR}/mlir" |
| TD_FILE dialects/NVGPUTransformOps.td |
| SOURCES |
| dialects/transform/nvgpu.py |
| DIALECT_NAME transform |
| EXTENSION_NAME nvgpu_transform) |
| |
| declare_mlir_dialect_extension_python_bindings( |
| ADD_TO_PARENT MLIRPythonSources.Dialects |
| ROOT_DIR "${CMAKE_CURRENT_SOURCE_DIR}/mlir" |
| TD_FILE dialects/LinalgStructuredTransformOps.td |
| SOURCES |
| dialects/transform/structured.py |
| DIALECT_NAME transform |
| EXTENSION_NAME structured_transform |
| GEN_ENUM_BINDINGS_TD_FILE |
| "../../include/mlir/Dialect/Linalg/TransformOps/LinalgTransformEnums.td" |
| ) |
| |
| declare_mlir_dialect_extension_python_bindings( |
| ADD_TO_PARENT MLIRPythonSources.Dialects |
| ROOT_DIR "${CMAKE_CURRENT_SOURCE_DIR}/mlir" |
| TD_FILE dialects/SparseTensorTransformOps.td |
| SOURCES |
| dialects/transform/sparse_tensor.py |
| DIALECT_NAME transform |
| EXTENSION_NAME sparse_tensor_transform) |
| |
| declare_mlir_dialect_extension_python_bindings( |
| ADD_TO_PARENT MLIRPythonSources.Dialects |
| ROOT_DIR "${CMAKE_CURRENT_SOURCE_DIR}/mlir" |
| TD_FILE dialects/TensorTransformOps.td |
| SOURCES |
| dialects/transform/tensor.py |
| DIALECT_NAME transform |
| EXTENSION_NAME tensor_transform) |
| |
| declare_mlir_dialect_extension_python_bindings( |
| ADD_TO_PARENT MLIRPythonSources.Dialects |
| ROOT_DIR "${CMAKE_CURRENT_SOURCE_DIR}/mlir" |
| TD_FILE dialects/VectorTransformOps.td |
| SOURCES |
| dialects/transform/vector.py |
| DIALECT_NAME transform |
| EXTENSION_NAME vector_transform |
| GEN_ENUM_BINDINGS_TD_FILE |
| "../../include/mlir/Dialect/Vector/Transforms/VectorTransformsBase.td" |
| ) |
| |
| declare_mlir_dialect_python_bindings( |
| ADD_TO_PARENT MLIRPythonSources.Dialects |
| ROOT_DIR "${CMAKE_CURRENT_SOURCE_DIR}/mlir" |
| TD_FILE dialects/MathOps.td |
| SOURCES dialects/math.py |
| DIALECT_NAME math) |
| |
| declare_mlir_dialect_python_bindings( |
| ADD_TO_PARENT MLIRPythonSources.Dialects |
| ROOT_DIR "${CMAKE_CURRENT_SOURCE_DIR}/mlir" |
| TD_FILE dialects/ArithOps.td |
| SOURCES |
| dialects/arith.py |
| DIALECT_NAME arith |
| GEN_ENUM_BINDINGS) |
| |
| declare_mlir_dialect_python_bindings( |
| ADD_TO_PARENT MLIRPythonSources.Dialects |
| ROOT_DIR "${CMAKE_CURRENT_SOURCE_DIR}/mlir" |
| TD_FILE dialects/MemRefOps.td |
| SOURCES |
| dialects/memref.py |
| DIALECT_NAME memref) |
| |
| declare_mlir_dialect_python_bindings( |
| ADD_TO_PARENT MLIRPythonSources.Dialects |
| ROOT_DIR "${CMAKE_CURRENT_SOURCE_DIR}/mlir" |
| TD_FILE dialects/MLProgramOps.td |
| SOURCES |
| dialects/ml_program.py |
| DIALECT_NAME ml_program) |
| |
| declare_mlir_dialect_python_bindings( |
| ADD_TO_PARENT MLIRPythonSources.Dialects |
| ROOT_DIR "${CMAKE_CURRENT_SOURCE_DIR}/mlir" |
| TD_FILE dialects/NVGPUOps.td |
| SOURCES |
| dialects/nvgpu.py |
| DIALECT_NAME nvgpu |
| GEN_ENUM_BINDINGS) |
| |
| declare_mlir_dialect_python_bindings( |
| ADD_TO_PARENT MLIRPythonSources.Dialects |
| ROOT_DIR "${CMAKE_CURRENT_SOURCE_DIR}/mlir" |
| TD_FILE dialects/NVVMOps.td |
| SOURCES |
| dialects/nvvm.py |
| DIALECT_NAME nvvm |
| GEN_ENUM_BINDINGS) |
| |
| declare_mlir_dialect_python_bindings( |
| ADD_TO_PARENT MLIRPythonSources.Dialects |
| ROOT_DIR "${CMAKE_CURRENT_SOURCE_DIR}/mlir" |
| TD_FILE dialects/ROCDLOps.td |
| SOURCES |
| dialects/rocdl.py |
| DIALECT_NAME rocdl) |
| |
| declare_mlir_python_sources( |
| MLIRPythonSources.Dialects.quant |
| ADD_TO_PARENT MLIRPythonSources.Dialects |
| ROOT_DIR "${CMAKE_CURRENT_SOURCE_DIR}/mlir" |
| GEN_ENUM_BINDINGS |
| SOURCES |
| dialects/quant.py |
| _mlir_libs/_mlir/dialects/quant.pyi) |
| |
| declare_mlir_dialect_python_bindings( |
| ADD_TO_PARENT MLIRPythonSources.Dialects |
| ROOT_DIR "${CMAKE_CURRENT_SOURCE_DIR}/mlir" |
| TD_FILE dialects/EmitC.td |
| SOURCES |
| dialects/emitc.py |
| DIALECT_NAME emitc) |
| |
| declare_mlir_dialect_python_bindings( |
| ADD_TO_PARENT MLIRPythonSources.Dialects |
| ROOT_DIR "${CMAKE_CURRENT_SOURCE_DIR}/mlir" |
| TD_FILE dialects/PDLOps.td |
| SOURCES |
| dialects/pdl.py |
| _mlir_libs/_mlir/dialects/pdl.pyi |
| DIALECT_NAME pdl) |
| |
| declare_mlir_dialect_python_bindings( |
| ADD_TO_PARENT MLIRPythonSources.Dialects |
| ROOT_DIR "${CMAKE_CURRENT_SOURCE_DIR}/mlir" |
| TD_FILE dialects/OpenMPOps.td |
| SOURCES |
| dialects/openmp.py |
| DIALECT_NAME omp |
| DEPENDS omp_common_td) |
| |
| declare_mlir_dialect_python_bindings( |
| ADD_TO_PARENT MLIRPythonSources.Dialects |
| ROOT_DIR "${CMAKE_CURRENT_SOURCE_DIR}/mlir" |
| TD_FILE dialects/SCFOps.td |
| SOURCES |
| dialects/scf.py |
| DIALECT_NAME scf) |
| |
| declare_mlir_dialect_python_bindings( |
| ADD_TO_PARENT MLIRPythonSources.Dialects |
| ROOT_DIR "${CMAKE_CURRENT_SOURCE_DIR}/mlir" |
| TD_FILE dialects/ShapeOps.td |
| SOURCES dialects/shape.py |
| DIALECT_NAME shape) |
| |
| declare_mlir_dialect_python_bindings( |
| ADD_TO_PARENT MLIRPythonSources.Dialects |
| ROOT_DIR "${CMAKE_CURRENT_SOURCE_DIR}/mlir" |
| TD_FILE dialects/SparseTensorOps.td |
| SOURCES dialects/sparse_tensor.py |
| DIALECT_NAME sparse_tensor |
| GEN_ENUM_BINDINGS_TD_FILE |
| "../../include/mlir/Dialect/SparseTensor/IR/SparseTensorAttrDefs.td" |
| ) |
| |
| declare_mlir_dialect_python_bindings( |
| ADD_TO_PARENT MLIRPythonSources.Dialects |
| ROOT_DIR "${CMAKE_CURRENT_SOURCE_DIR}/mlir" |
| TD_FILE dialects/SMTOps.td |
| GEN_ENUM_BINDINGS |
| SOURCES |
| dialects/smt.py |
| DIALECT_NAME smt) |
| |
| declare_mlir_dialect_python_bindings( |
| ADD_TO_PARENT MLIRPythonSources.Dialects |
| ROOT_DIR "${CMAKE_CURRENT_SOURCE_DIR}/mlir" |
| TD_FILE dialects/SPIRVOps.td |
| SOURCES dialects/spirv.py |
| DIALECT_NAME spirv) |
| |
| declare_mlir_dialect_python_bindings( |
| ADD_TO_PARENT MLIRPythonSources.Dialects |
| ROOT_DIR "${CMAKE_CURRENT_SOURCE_DIR}/mlir" |
| TD_FILE dialects/TensorOps.td |
| SOURCES |
| dialects/tensor.py |
| DIALECT_NAME tensor) |
| |
| declare_mlir_dialect_python_bindings( |
| ADD_TO_PARENT MLIRPythonSources.Dialects |
| ROOT_DIR "${CMAKE_CURRENT_SOURCE_DIR}/mlir" |
| TD_FILE dialects/TosaOps.td |
| SOURCES dialects/tosa.py |
| DIALECT_NAME tosa |
| ) |
| |
| declare_mlir_dialect_python_bindings( |
| ADD_TO_PARENT MLIRPythonSources.Dialects |
| ROOT_DIR "${CMAKE_CURRENT_SOURCE_DIR}/mlir" |
| TD_FILE dialects/VectorOps.td |
| SOURCES dialects/vector.py |
| DIALECT_NAME vector |
| GEN_ENUM_BINDINGS_TD_FILE |
| "dialects/VectorAttributes.td") |
| |
| ################################################################################ |
| # Python extensions. |
| # The sources for these are all in lib/Bindings/Python, but since they have to |
| # be rebuilt for each package and integrate with the source setup here, we |
| # just reference them here instead of having ordered, cross package target |
| # dependencies. |
| ################################################################################ |
| |
| set(PYTHON_SOURCE_DIR "${MLIR_SOURCE_DIR}/lib/Bindings/Python") |
| declare_mlir_python_extension(MLIRPythonExtension.Core |
| MODULE_NAME _mlir |
| ADD_TO_PARENT MLIRPythonSources.Core |
| ROOT_DIR "${PYTHON_SOURCE_DIR}" |
| PYTHON_BINDINGS_LIBRARY nanobind |
| SOURCES |
| MainModule.cpp |
| IRAffine.cpp |
| IRAttributes.cpp |
| IRCore.cpp |
| IRInterfaces.cpp |
| IRModule.cpp |
| IRTypes.cpp |
| Pass.cpp |
| Rewrite.cpp |
| |
| # Headers must be included explicitly so they are installed. |
| Globals.h |
| IRModule.h |
| Pass.h |
| NanobindUtils.h |
| Rewrite.h |
| PRIVATE_LINK_LIBS |
| LLVMSupport |
| EMBED_CAPI_LINK_LIBS |
| MLIRCAPIDebug |
| MLIRCAPIIR |
| MLIRCAPIInterfaces |
| |
| # Dialects |
| MLIRCAPIFunc |
| ) |
| |
| # This extension exposes an API to register all dialects, extensions, and passes |
| # packaged in upstream MLIR and it is used for the upstream "mlir" Python |
| # package. Downstreams will likely want to provide their own and not depend |
| # on this one, since it links in the world. |
| # Note that this is not added to any top-level source target for transitive |
| # inclusion: It must be included explicitly by downstreams if desired. Note that |
| # this has a very large impact on what gets built/packaged. |
| declare_mlir_python_extension(MLIRPythonExtension.RegisterEverything |
| MODULE_NAME _mlirRegisterEverything |
| ROOT_DIR "${PYTHON_SOURCE_DIR}" |
| PYTHON_BINDINGS_LIBRARY nanobind |
| SOURCES |
| RegisterEverything.cpp |
| PRIVATE_LINK_LIBS |
| LLVMSupport |
| EMBED_CAPI_LINK_LIBS |
| MLIRCAPIConversion |
| MLIRCAPITransforms |
| MLIRCAPIRegisterEverything |
| ) |
| |
| declare_mlir_python_extension(MLIRPythonExtension.Dialects.Linalg.Pybind |
| MODULE_NAME _mlirDialectsLinalg |
| ADD_TO_PARENT MLIRPythonSources.Dialects.linalg |
| ROOT_DIR "${PYTHON_SOURCE_DIR}" |
| PYTHON_BINDINGS_LIBRARY nanobind |
| SOURCES |
| DialectLinalg.cpp |
| PRIVATE_LINK_LIBS |
| LLVMSupport |
| EMBED_CAPI_LINK_LIBS |
| MLIRCAPIIR |
| MLIRCAPILinalg |
| ) |
| |
| declare_mlir_python_extension(MLIRPythonExtension.Dialects.GPU.Pybind |
| MODULE_NAME _mlirDialectsGPU |
| ADD_TO_PARENT MLIRPythonSources.Dialects.gpu |
| ROOT_DIR "${PYTHON_SOURCE_DIR}" |
| PYTHON_BINDINGS_LIBRARY nanobind |
| SOURCES |
| DialectGPU.cpp |
| PRIVATE_LINK_LIBS |
| LLVMSupport |
| EMBED_CAPI_LINK_LIBS |
| MLIRCAPIIR |
| MLIRCAPIGPU |
| ) |
| |
| declare_mlir_python_extension(MLIRPythonExtension.Dialects.LLVM.Pybind |
| MODULE_NAME _mlirDialectsLLVM |
| ADD_TO_PARENT MLIRPythonSources.Dialects.llvm |
| ROOT_DIR "${PYTHON_SOURCE_DIR}" |
| PYTHON_BINDINGS_LIBRARY nanobind |
| SOURCES |
| DialectLLVM.cpp |
| PRIVATE_LINK_LIBS |
| LLVMSupport |
| EMBED_CAPI_LINK_LIBS |
| MLIRCAPIIR |
| MLIRCAPILLVM |
| ) |
| |
| declare_mlir_python_extension(MLIRPythonExtension.Dialects.Quant.Pybind |
| MODULE_NAME _mlirDialectsQuant |
| ADD_TO_PARENT MLIRPythonSources.Dialects.quant |
| ROOT_DIR "${PYTHON_SOURCE_DIR}" |
| PYTHON_BINDINGS_LIBRARY nanobind |
| SOURCES |
| DialectQuant.cpp |
| PRIVATE_LINK_LIBS |
| LLVMSupport |
| EMBED_CAPI_LINK_LIBS |
| MLIRCAPIIR |
| MLIRCAPIQuant |
| ) |
| |
| declare_mlir_python_extension(MLIRPythonExtension.Dialects.NVGPU.Pybind |
| MODULE_NAME _mlirDialectsNVGPU |
| ADD_TO_PARENT MLIRPythonSources.Dialects.nvgpu |
| ROOT_DIR "${PYTHON_SOURCE_DIR}" |
| PYTHON_BINDINGS_LIBRARY nanobind |
| SOURCES |
| DialectNVGPU.cpp |
| PRIVATE_LINK_LIBS |
| LLVMSupport |
| EMBED_CAPI_LINK_LIBS |
| MLIRCAPIIR |
| MLIRCAPINVGPU |
| ) |
| |
| declare_mlir_python_extension(MLIRPythonExtension.Dialects.PDL.Pybind |
| MODULE_NAME _mlirDialectsPDL |
| ADD_TO_PARENT MLIRPythonSources.Dialects.pdl |
| ROOT_DIR "${PYTHON_SOURCE_DIR}" |
| PYTHON_BINDINGS_LIBRARY nanobind |
| SOURCES |
| DialectPDL.cpp |
| PRIVATE_LINK_LIBS |
| LLVMSupport |
| EMBED_CAPI_LINK_LIBS |
| MLIRCAPIIR |
| MLIRCAPIPDL |
| ) |
| |
| declare_mlir_python_extension(MLIRPythonExtension.Dialects.SparseTensor.Pybind |
| MODULE_NAME _mlirDialectsSparseTensor |
| ADD_TO_PARENT MLIRPythonSources.Dialects.sparse_tensor |
| ROOT_DIR "${PYTHON_SOURCE_DIR}" |
| PYTHON_BINDINGS_LIBRARY nanobind |
| SOURCES |
| DialectSparseTensor.cpp |
| PRIVATE_LINK_LIBS |
| LLVMSupport |
| EMBED_CAPI_LINK_LIBS |
| MLIRCAPIIR |
| MLIRCAPISparseTensor |
| ) |
| |
| declare_mlir_python_extension(MLIRPythonExtension.Dialects.Transform.Pybind |
| MODULE_NAME _mlirDialectsTransform |
| ADD_TO_PARENT MLIRPythonSources.Dialects.transform |
| ROOT_DIR "${PYTHON_SOURCE_DIR}" |
| PYTHON_BINDINGS_LIBRARY nanobind |
| SOURCES |
| DialectTransform.cpp |
| PRIVATE_LINK_LIBS |
| LLVMSupport |
| EMBED_CAPI_LINK_LIBS |
| MLIRCAPIIR |
| MLIRCAPITransformDialect |
| ) |
| |
| declare_mlir_python_extension(MLIRPythonExtension.AsyncDialectPasses |
| MODULE_NAME _mlirAsyncPasses |
| ADD_TO_PARENT MLIRPythonSources.Dialects.async |
| ROOT_DIR "${PYTHON_SOURCE_DIR}" |
| PYTHON_BINDINGS_LIBRARY nanobind |
| SOURCES |
| AsyncPasses.cpp |
| PRIVATE_LINK_LIBS |
| LLVMSupport |
| EMBED_CAPI_LINK_LIBS |
| MLIRCAPIAsync |
| ) |
| |
| if(MLIR_ENABLE_EXECUTION_ENGINE) |
| declare_mlir_python_extension(MLIRPythonExtension.ExecutionEngine |
| MODULE_NAME _mlirExecutionEngine |
| ADD_TO_PARENT MLIRPythonSources.ExecutionEngine |
| ROOT_DIR "${PYTHON_SOURCE_DIR}" |
| PYTHON_BINDINGS_LIBRARY nanobind |
| SOURCES |
| ExecutionEngineModule.cpp |
| PRIVATE_LINK_LIBS |
| LLVMSupport |
| EMBED_CAPI_LINK_LIBS |
| MLIRCAPIExecutionEngine |
| ) |
| endif() |
| |
| declare_mlir_python_extension(MLIRPythonExtension.GPUDialectPasses |
| MODULE_NAME _mlirGPUPasses |
| ADD_TO_PARENT MLIRPythonSources.Dialects.gpu |
| ROOT_DIR "${PYTHON_SOURCE_DIR}" |
| PYTHON_BINDINGS_LIBRARY nanobind |
| SOURCES |
| GPUPasses.cpp |
| PRIVATE_LINK_LIBS |
| LLVMSupport |
| EMBED_CAPI_LINK_LIBS |
| MLIRCAPIGPU |
| ) |
| |
| declare_mlir_python_extension(MLIRPythonExtension.LinalgPasses |
| MODULE_NAME _mlirLinalgPasses |
| ADD_TO_PARENT MLIRPythonSources.Dialects.linalg |
| ROOT_DIR "${PYTHON_SOURCE_DIR}" |
| PYTHON_BINDINGS_LIBRARY nanobind |
| SOURCES |
| LinalgPasses.cpp |
| PRIVATE_LINK_LIBS |
| LLVMSupport |
| EMBED_CAPI_LINK_LIBS |
| MLIRCAPILinalg |
| ) |
| |
| declare_mlir_python_extension(MLIRPythonExtension.Dialects.SMT.Pybind |
| MODULE_NAME _mlirDialectsSMT |
| ADD_TO_PARENT MLIRPythonSources.Dialects.smt |
| ROOT_DIR "${PYTHON_SOURCE_DIR}" |
| PYTHON_BINDINGS_LIBRARY nanobind |
| SOURCES |
| DialectSMT.cpp |
| # Headers must be included explicitly so they are installed. |
| NanobindUtils.h |
| PRIVATE_LINK_LIBS |
| LLVMSupport |
| EMBED_CAPI_LINK_LIBS |
| MLIRCAPIIR |
| MLIRCAPISMT |
| MLIRCAPIExportSMTLIB |
| ) |
| |
| declare_mlir_python_extension(MLIRPythonExtension.SparseTensorDialectPasses |
| MODULE_NAME _mlirSparseTensorPasses |
| ADD_TO_PARENT MLIRPythonSources.Dialects.sparse_tensor |
| ROOT_DIR "${PYTHON_SOURCE_DIR}" |
| PYTHON_BINDINGS_LIBRARY nanobind |
| SOURCES |
| SparseTensorPasses.cpp |
| PRIVATE_LINK_LIBS |
| LLVMSupport |
| EMBED_CAPI_LINK_LIBS |
| MLIRCAPISparseTensor |
| ) |
| |
| declare_mlir_python_extension(MLIRPythonExtension.TransformInterpreter |
| MODULE_NAME _mlirTransformInterpreter |
| ADD_TO_PARENT MLIRPythonSources.Dialects.transform |
| ROOT_DIR "${PYTHON_SOURCE_DIR}" |
| PYTHON_BINDINGS_LIBRARY nanobind |
| SOURCES |
| TransformInterpreter.cpp |
| PRIVATE_LINK_LIBS |
| LLVMSupport |
| EMBED_CAPI_LINK_LIBS |
| MLIRCAPITransformDialectTransforms |
| ) |
| |
| # TODO: Figure out how to put this in the test tree. |
| # This should not be included in the main Python extension. However, |
| # putting it into MLIRPythonTestSources along with the dialect declaration |
| # above confuses Python module loader when running under lit. |
| set(_ADDL_TEST_SOURCES) |
| if(MLIR_INCLUDE_TESTS) |
| set(_ADDL_TEST_SOURCES MLIRPythonTestSources) |
| declare_mlir_python_sources(MLIRPythonTestSources) |
| declare_mlir_python_sources(MLIRPythonTestSources.Dialects |
| ADD_TO_PARENT MLIRPythonTestSources) |
| |
| # TODO: this uses a tablegen file from the test directory and should be |
| # decoupled from here. |
| declare_mlir_python_sources( |
| MLIRPythonTestSources.Dialects.PythonTest |
| ROOT_DIR "${CMAKE_CURRENT_SOURCE_DIR}/mlir" |
| ADD_TO_PARENT MLIRPythonTestSources.Dialects |
| SOURCES |
| dialects/python_test.py |
| ) |
| set(LLVM_TARGET_DEFINITIONS |
| "${MLIR_MAIN_SRC_DIR}/test/python/python_test_ops.td") |
| mlir_tablegen( |
| "dialects/_python_test_ops_gen.py" |
| -gen-python-op-bindings |
| -bind-dialect=python_test) |
| add_public_tablegen_target(PythonTestDialectPyIncGen) |
| declare_mlir_python_sources( |
| MLIRPythonTestSources.Dialects.PythonTest.ops_gen |
| ROOT_DIR "${CMAKE_CURRENT_BINARY_DIR}" |
| ADD_TO_PARENT MLIRPythonTestSources.Dialects.PythonTest |
| SOURCES "dialects/_python_test_ops_gen.py") |
| |
| declare_mlir_python_extension(MLIRPythonTestSources.PythonTestExtensionPybind11 |
| MODULE_NAME _mlirPythonTestPybind11 |
| ADD_TO_PARENT MLIRPythonTestSources.Dialects |
| ROOT_DIR "${MLIR_SOURCE_DIR}/test/python/lib" |
| PYTHON_BINDINGS_LIBRARY pybind11 |
| SOURCES |
| PythonTestModulePybind11.cpp |
| PRIVATE_LINK_LIBS |
| LLVMSupport |
| EMBED_CAPI_LINK_LIBS |
| MLIRCAPIPythonTestDialect |
| ) |
| declare_mlir_python_extension(MLIRPythonTestSources.PythonTestExtensionNanobind |
| MODULE_NAME _mlirPythonTestNanobind |
| ADD_TO_PARENT MLIRPythonTestSources.Dialects |
| ROOT_DIR "${MLIR_SOURCE_DIR}/test/python/lib" |
| PYTHON_BINDINGS_LIBRARY nanobind |
| SOURCES |
| PythonTestModuleNanobind.cpp |
| PRIVATE_LINK_LIBS |
| LLVMSupport |
| EMBED_CAPI_LINK_LIBS |
| MLIRCAPIPythonTestDialect |
| ) |
| endif() |
| |
| ################################################################################ |
| # Common CAPI dependency DSO. |
| # All python extensions must link through one DSO which exports the CAPI, and |
| # this must have a globally unique name amongst all embeddors of the python |
| # library since it will effectively have global scope. |
| # |
| # The presence of this aggregate library is part of the long term plan, but its |
| # use needs to be made more flexible. |
| # |
| # TODO: Upgrade to the aggregate utility in https://reviews.llvm.org/D106419 |
| # once ready. |
| ################################################################################ |
| |
| add_mlir_python_common_capi_library(MLIRPythonCAPI |
| INSTALL_COMPONENT MLIRPythonModules |
| INSTALL_DESTINATION "${MLIR_BINDINGS_PYTHON_INSTALL_PREFIX}/_mlir_libs" |
| OUTPUT_DIRECTORY "${MLIR_BINARY_DIR}/python_packages/mlir_core/mlir/_mlir_libs" |
| RELATIVE_INSTALL_ROOT "../../../.." |
| DECLARED_HEADERS |
| MLIRPythonCAPI.HeaderSources |
| DECLARED_SOURCES |
| MLIRPythonSources |
| MLIRPythonExtension.RegisterEverything |
| ${_ADDL_TEST_SOURCES} |
| ) |
| |
| ################################################################################ |
| # Custom targets. |
| ################################################################################ |
| |
| _flatten_mlir_python_targets(mlir_python_sources_deps MLIRPythonSources) |
| add_custom_target("mlir-python-sources" DEPENDS ${mlir_python_sources_deps}) |
| if(NOT LLVM_ENABLE_IDE) |
| add_llvm_install_targets(install-mlir-python-sources |
| DEPENDS mlir-python-sources |
| COMPONENT mlir-python-sources |
| ) |
| endif() |
| |
| ################################################################################ |
| # The fully assembled package of modules. |
| # This must come last. |
| ################################################################################ |
| |
| add_mlir_python_modules(MLIRPythonModules |
| ROOT_PREFIX "${MLIR_BINARY_DIR}/python_packages/mlir_core/mlir" |
| INSTALL_PREFIX "${MLIR_BINDINGS_PYTHON_INSTALL_PREFIX}" |
| DECLARED_SOURCES |
| MLIRPythonSources |
| MLIRPythonExtension.RegisterEverything |
| ${_ADDL_TEST_SOURCES} |
| COMMON_CAPI_LINK_LIBS |
| MLIRPythonCAPI |
| ) |
| |
