392622d084
Reverts revert #118517 after (hopefully) fixing builders
(https://github.com/llvm/llvm-zorg/pull/328,
https://github.com/llvm/llvm-zorg/pull/327)
This reverts commit 61bf308cf2.
79 lines
2.7 KiB
CMake
79 lines
2.7 KiB
CMake
include(AddMLIRPython)
|
|
|
|
# Specifies that all MLIR packages are co-located under the `mlir_standalone`
|
|
# top level package (the API has been embedded in a relocatable way).
|
|
# TODO: Add an upstream cmake param for this vs having a global here.
|
|
add_compile_definitions("MLIR_PYTHON_PACKAGE_PREFIX=mlir_standalone.")
|
|
|
|
|
|
################################################################################
|
|
# Sources
|
|
################################################################################
|
|
|
|
declare_mlir_python_sources(StandalonePythonSources)
|
|
|
|
declare_mlir_dialect_python_bindings(
|
|
ADD_TO_PARENT StandalonePythonSources
|
|
ROOT_DIR "${CMAKE_CURRENT_SOURCE_DIR}/mlir_standalone"
|
|
TD_FILE dialects/StandaloneOps.td
|
|
SOURCES
|
|
dialects/standalone_pybind11.py
|
|
dialects/standalone_nanobind.py
|
|
DIALECT_NAME standalone)
|
|
|
|
|
|
declare_mlir_python_extension(StandalonePythonSources.Pybind11Extension
|
|
MODULE_NAME _standaloneDialectsPybind11
|
|
ADD_TO_PARENT StandalonePythonSources
|
|
SOURCES
|
|
StandaloneExtensionPybind11.cpp
|
|
EMBED_CAPI_LINK_LIBS
|
|
StandaloneCAPI
|
|
PYTHON_BINDINGS_LIBRARY pybind11
|
|
)
|
|
|
|
declare_mlir_python_extension(StandalonePythonSources.NanobindExtension
|
|
MODULE_NAME _standaloneDialectsNanobind
|
|
ADD_TO_PARENT StandalonePythonSources
|
|
SOURCES
|
|
StandaloneExtensionNanobind.cpp
|
|
EMBED_CAPI_LINK_LIBS
|
|
StandaloneCAPI
|
|
PYTHON_BINDINGS_LIBRARY nanobind
|
|
)
|
|
|
|
|
|
################################################################################
|
|
# Common CAPI
|
|
################################################################################
|
|
|
|
add_mlir_python_common_capi_library(StandalonePythonCAPI
|
|
INSTALL_COMPONENT StandalonePythonModules
|
|
INSTALL_DESTINATION python_packages/standalone/mlir_standalone/_mlir_libs
|
|
OUTPUT_DIRECTORY "${MLIR_BINARY_DIR}/python_packages/standalone/mlir_standalone/_mlir_libs"
|
|
RELATIVE_INSTALL_ROOT "../../../.."
|
|
DECLARED_SOURCES
|
|
StandalonePythonSources
|
|
# TODO: Remove this in favor of showing fine grained registration once
|
|
# available.
|
|
MLIRPythonExtension.RegisterEverything
|
|
MLIRPythonSources.Core
|
|
)
|
|
|
|
################################################################################
|
|
# Instantiation of all Python modules
|
|
################################################################################
|
|
|
|
add_mlir_python_modules(StandalonePythonModules
|
|
ROOT_PREFIX "${MLIR_BINARY_DIR}/python_packages/standalone/mlir_standalone"
|
|
INSTALL_PREFIX "python_packages/standalone/mlir_standalone"
|
|
DECLARED_SOURCES
|
|
StandalonePythonSources
|
|
# TODO: Remove this in favor of showing fine grained registration once
|
|
# available.
|
|
MLIRPythonExtension.RegisterEverything
|
|
MLIRPythonSources
|
|
COMMON_CAPI_LINK_LIBS
|
|
StandalonePythonCAPI
|
|
)
|